【C - TT 的美梦】spfa思想应用

题意：

有 N 个商业城市，编号 1 ～ N，其中 1 号城市是首都。共有 M 条有向道路供商业城市相互往来。对每一个商业城市标记一个正整数，表示其繁荣程度，当从一个商业城市走到另一个商业城市时，收取（目的地繁荣程度 - 出发地繁荣程度）^ 3 的税。
从首都出发，走到其他城市至少要交多少的税，如果总金额小于 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:
?
?

思路：

将边权直接设为（目的地繁荣程度 - 出发地繁荣程度）^ 3，则会出现负数，那么很有可能出现负环，造成死循环。从一个点到另一个点的最短路程经过的边不可能超过n-1条，所以用一个数组记录从起点到该点经过的边数，如果大于等于n说明陷入了负环，为其打上标签，不再使用该点，并且利用dfs，遍历其能到达的所有点，都打上负环标签。

总结：

在spfa中，一个点可以多次入队。

代码：

#include<iostream>
#include<cstring>
#include<queue>
#include<cmath>
using namespace std;
const int inf = 999999999;
struct edge
{
	int next;
	int to;
	int w;
}Edge[100010];
int a[205], head[205], neg[205], dis[205], cnt[205], vis[205];
int tot;
void add_edge(int u, int v, int w)
{
	Edge[++tot].to = v;
	Edge[tot].w = w;
	Edge[tot].next = head[u];
	head[u] = tot;
}
void dfs(int x)
{
	for (int i = head[x]; i != 0; i = Edge[i].next)
	{
		int t = Edge[i].to;
		if (!neg[t])
		{
			neg[t] = 1;
			dfs(t);
		}
	}
}
void spfa(int n)
{
	queue<int> Q;
	dis[1] = 0;
	Q.push(1);
	while (!Q.empty())
	{
		int u = Q.front(); 
		Q.pop();
		vis[u] = 0;//判断当前点是否在队列中
		for (int i = head[u]; i != 0; i = Edge[i].next)
		{
			int v = Edge[i].to, w = Edge[i].w;
			if (neg[v]) continue;
			if (dis[v] > dis[u] + w)
			{
				dis[v] = dis[u] + w;
				cnt[v] = cnt[u] + 1;
				if (cnt[v] >= n)//负环
				{
					neg[v] = 1;
					dfs(v);
				}
				if (!vis[v])
				{
					Q.push(v);
					vis[v] = 1;
				}
			}
		}
	}
}
void ini(int n)
{
	tot = 0;
	memset(a, 0, sizeof(a));
	memset(head, 0, sizeof(head));
	memset(neg, 0, sizeof(neg));
	memset(vis, 0, sizeof(vis));
	memset(cnt, 0, sizeof(cnt));
	for (int i = 1; i <= n; i++)
		dis[i] = inf;
}
int main()
{
	int T;
	cin >> T;
	int inx = 1;
	while (T--)
	{
		int N, M, Q;
		cin >> N;
		ini(N);
		for (int i = 1; i <= N; i++)
			cin >> a[i];
		cin >> M;
		while (M--)
		{
			int ia, b;
			cin >> ia;
			cin >> b;
			add_edge(ia, b, pow(a[b] - a[ia], 3));
		}
		spfa(N);
		cout << "Case " << inx++ << ":" << endl;
		cin >> Q;
		while (Q--)
		{
			int p;
			cin >> p;
			if (neg[p] == 1 || dis[p] == inf || dis[p] < 3)
				cout << "?" << endl;
			else
				cout << dis[p] << endl;
		}
	}
}