LightOJ-1074 Extended Traffic

题目链接：https://vjudge.net/problem/LightOJ-1074

最短路问题，而且会有负环，需要把负环上的点都找出来，用spfa+dfs

用stack实现的spfa，比起用queue实现的快了有一倍（有负环情况下stack更快）

这题卡了很久，因为一个莫名的地方（代码里用注释标出来了），心累。。。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<stack>
using namespace std;
const int N=200+20;
const int inf=0x3f3f3f3f;
int g[N][N];
bool vis[N],r[N],g0[N][N];
int d[N],c[N];
int t,n,m,qu;
void dfs(int u) //寻找负环并标记 
{
	r[u]=true;
	for(int i=1;i<=n;i++)
	if(g0[u][i]&&!r[i])
	dfs(i);
}
int main()
{
	scanf("%d",&t);
	int kase=0;
	while(t--)
	{
		int num[N],cnt[N],a,b;
		memset(g,inf,sizeof(g));
		memset(g0,false,sizeof(g0));
		scanf("%d",&n);
		for(int i=1;i<=n;i++)
		scanf("%d",&num[i]);
		scanf("%d",&m);
		for(int i=0;i<m;i++)
		{
			scanf("%d%d",&a,&b);
			g[a][b]=(num[b]-num[a])*(num[b]-num[a])*(num[b]-num[a]);
			g0[a][b]=true;
		}
		stack<int> q;
		q.push(1);
		memset(d,inf,sizeof(d));
		memset(vis,false,sizeof(vis));
		memset(r,false,sizeof(r));
		memset(cnt,0,sizeof(cnt));
		d[1]=0;vis[1]=true;cnt[1]=1;
		while(!q.empty())  //spfa 
		{
			int u=q.top();
			q.pop();vis[u]=false;
			for(int i=1;i<=n;i++)
			if(!r[i]&&g0[u][i]&&d[i]>d[u]+g[u][i]) //必须先判断g0[u][i]是否联通？？？不然WA！！！ 
			{
				d[i]=d[u]+g[u][i];
				if(!vis[i])
				{
					q.push(i);
					vis[i]=true;
					cnt[i]++;
					if(cnt[i]>n)
					dfs(i);
				}
			}
		}
		scanf("%d",&qu);
		for(int i=0;i<qu;i++)
		scanf("%d",&c[i]);
		printf("Case %d:\n",++kase);
		for(int i=0;i<qu;i++)
		{
			a=c[i];
			if(d[a]<3||d[a]>=inf||r[a]) printf("?\n");
			else printf("%d\n",d[a]);
		}
	}
	return 0;
}