POJ 3259_SPFA_判断是否含负权环

题目链接：http://poj.org/problem?id=3259

#include<iostream>
using namespace std;
#include<vector>
#include<queue>
#include<algorithm>
#include<cstring>
int F,N,M,W;
const int INF=1<<30;
struct Edge
{
    int e,w;
    Edge(int ee,int ww):e(ee),w(ww) {}
    Edge() {}
};
vector<Edge> G[1010];
int updateTimes[1010];
int dist[1010];
int spfa(int v)
{
    for (int i=1;i<=N;i++)
        dist[i]=INF;
    dist[v]=0;
    queue<int>que;
    que.push(v);
    memset(updateTimes,0,sizeof(updateTimes));
    while (!que.empty())
    {
        int s=que.front();
        que.pop();
        for (int i=0;i<G[s].size();i++)
        {
            int e=G[s][i].e;
            if (dist[e]>dist[s]+G[s][i].w)
            {
                dist[e]=dist[s]+G[s][i].w;
                que.push(e);
                updateTimes[e]++;
                if (updateTimes[e]>=N)
                    return true;
            }
        }
    }
    return false;
}
int main()
{
    cin>>F;
    while(F--)
    {
        cin>>N>>M>>W;
        for (int i=1;i<1000;i++)
            G[i].clear();
        int s,e,t;
        for (int i=0;i<M;i++)
        {
            cin>>s>>e>>t;
            G[s].push_back(Edge(e,t));
            G[e].push_back(Edge(s,t));
        }
        for (int i=0;i<W;i++)
        {
            cin>>s>>e>>t;
            G[s].push_back(Edge(e,-t));
        }
        if (spfa(1))
            cout<<"YES"<<endl;
        else
            cout<<"NO"<<endl;
    }
    return 0;
}