SDU_week7_C - TT 的美梦（前向星存图+带负环的SPFA+dfs打标）

题目描述

这一晚，TT 做了个美梦！

在梦中，TT 的愿望成真了，他成为了喵星的统领！喵星上有 N 个商业城市，编号 1 ～ N，其中 1 号城市是 TT 所在的城市，即首都。

喵星上共有 M 条有向道路供商业城市相互往来。但是随着喵星商业的日渐繁荣，有些道路变得非常拥挤。正在 TT 为之苦恼之时，他的魔法小猫咪提出了一个解决方案！TT 欣然接受并针对该方案颁布了一项新的政策。

具体政策如下：对每一个商业城市标记一个正整数，表示其繁荣程度，当每一只喵沿道路从一个商业城市走到另一个商业城市时，TT 都会收取它们（目的地繁荣程度 - 出发地繁荣程度）^ 3 的税。

TT 打算测试一下这项政策是否合理，因此他想知道从首都出发，走到其他城市至少要交多少的税，如果总金额小于 3 或者无法到达请悄咪咪地打出 ‘?’。

Input
第一行输入 T，表明共有 T 组数据。（1 <= T <= 50）
对于每一组数据，第一行输入 N，表示点的个数。（1 <= N <= 200）
第二行输入 N 个整数，表示 1 ～ N 点的权值 a[i]。（0 <= a[i] <= 20）
第三行输入 M，表示有向道路的条数。（0 <= M <= 100000）
接下来 M 行，每行有两个整数 A B，表示存在一条 A 到 B 的有向道路。
接下来给出一个整数 Q，表示询问个数。（0 <= Q <= 100000）
每一次询问给出一个 P，表示求 1 号点到 P 号点的最少税费。

Output
每个询问输出一行，如果不可达或税费小于 3 则输出 ‘?’。

Sample Input
2
5
6 7 8 9 10
6
1 2
2 3
3 4
1 5
5 4
4 5
2
4
5
10
1 2 4 4 5 6 7 8 9 10
10
1 2
2 3
3 1
1 4
4 5
5 6
6 7
7 8
8 9
9 10
2
3 10

Sample Output
Case 1:
3
4
Case 2:
?
?

题目分析

这道题的大意是给出带负权的有向图（边权通过点权运算获得），通过SPFA求单源最短路径，再根据条件输出即可。
不能正确输出（即输出?）有以下几种情况：

1. 不可到达。此时dis[i]根本没有被更新过，里面存的是inf
2. 权值小于3。即dis[i] < 3
3. 负环上的点。这些点最短路必然小于3，因为是负无穷，是第二种的特殊情况。如果负环上不及时终止，则SPFA会一直循环执行下去，最终导致RE。因此我们记录从源点到达某点路径长度，大于等于N时则在这个点进行dfs操作，使得这一些列负无穷的点被数组inf[]标记。 最后判断这个点的inf[]是否被置1即可知道是不是负环上的点可到达的点。

代码

#define _CRT_SECURE_NO_WARNINGS
#define _ ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);

#include <bits/stdc++.h>
//#include <iostream>
//#include<algorithm>

using namespace std;
int T,N,M,Q;//T组数据，N个点，M条边，Q询问次数
int point[209];//点权

struct edge
{
	int u, v, w, next;
}e[100009];
int tot;//边数
int head[100009];
void addEdge(int u,int v,int w)
{
	e[++tot].u = u;
	e[tot].v = v;
	e[tot].w = w;
	e[tot].next = head[u];
	head[u] = tot;
}

bool inf[209];
void dfs(int u)
{//将负权环路上的点能到达的所有点标记
	for (int i = head[u]; i; i = e[i].next)
	{
		int v = e[i].v;
		if (!inf[v])
		{
			inf[v] = true;
			dfs(v);
		}
	}
}

int parent[209];//父节点
int dis[209];//单源最短路径
int inq[209];//是否已经入队
int cnt[209];//到达某点路径长度
void SPFA(int s)
{
	queue<int> q;
	memset(parent, 0, sizeof(parent));
	memset(dis, 0x3f, sizeof(dis));
	memset(inq, 0, sizeof(inq));
	memset(cnt, 0, sizeof(cnt));
	
	dis[s] = 0;
	inq[s] = 1;
	q.push(s);
	while (!q.empty())
	{
		int u = q.front(); q.pop();
		inq[u] = 0;
		for (int i = head[u]; i ; i=e[i].next)
		{
			int v = e[i].v;
			int w = e[i].w;
			if (dis[v] > dis[u] + w)
			{
				cnt[v] = cnt[u] + 1;
				if (cnt[v] >= N)
				{//从源点到达某点路径长度大于等于N
					dfs(v);
					continue;
				}

				dis[v] = dis[u] + w;
				parent[v] = u;
				if (!inq[v])
				{
					q.push(v);
					inq[v] = 1;
				}
			}
		}
	}
}

int main()
{
    cin >> T;
	for(int i=1;i<=T;i++)
	{
		memset(point, 0, sizeof(point));
		memset(head, 0, sizeof(head));
		memset(inf, 0, sizeof(inf));
		cin >> N;
		for (int i = 1; i <= N; i++)cin >> point[i];

		cin >> M;
		tot = 0;
		for (int i = 1; i <= M; i++)
		{
			int u, v;
			cin >> u >> v;
			addEdge(u, v, pow(point[v] - point[u], 3));
		}

		SPFA(1);

		cout << "Case " << i << ":" << endl;
		cin >> Q;
		for (int i = 1; i <= Q; i++)
		{
			int temp;
			cin >> temp;
			if (inf[temp] || dis[temp] < 3 || dis[temp]>99999)
				cout << '?' << endl;
			else cout << dis[temp] << endl;
		}
	}
    return 0;
}