poj3259（spfa判负环）

 

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

题意：John的农场里N块地，M条路连接两块地，W个虫洞，虫洞是一条单向路，会在你离开之前把你传送到目的地，就是当你过去的时候时间会倒退Ts。我们的任务是知道会不会在从某块地出发后又回来，看到了离开之前的自己。总的来说，就是看图中有没有负权环。有的话就是可以，没有的话就是不可以了。

分析：sfa判负环，直接建图套模板即可。

#include <cstdio>

#include <cstring>

#include <string>

#include <cmath>

#include <iostream>

#include <algorithm>

#include <queue>

#include <cstdlib>

#include <stack>

#include <vector>

#include <set>

#include <map>

#define LL long long

#define mod 100000000

#define inf 0x3f3f3f3f

#define eps 1e-9

#define N 3010

#define FILL(a,b) (memset(a,b,sizeof(a)))

#define lson l,m,rt<<1

#define rson m+1,r,rt<<1|1

#define PII pair<int,int>

using namespace std;

struct edge

{

    int v,w,next;

    edge(){}

    edge(int v,int w,int next):v(v),w(w),next(next){}

}e[N<<1];

int head[N<<1],tot;

int vis[N],cnt[N];

int n,m,s;

int dis[N];

void init()

{

    FILL(vis,0);FILL(head,-1);

    FILL(dis,0x3f);FILL(cnt,0);tot=0;

}

void addedge(int u,int v,int w)

{

    e[tot]=edge(v,w,head[u]);

    head[u]=tot++;

}

int spfa()

{

    queue<int>que;

    vis[1]=1;cnt[1]=1;

    dis[1]=0;que.push(1);

    while(!que.empty())

    {

        int u=que.front();

        que.pop();vis[u]=0;

        for(int i=head[u];~i;i=e[i].next)

        {

            int v=e[i].v,w=e[i].w;

            if(dis[v]>dis[u]+w)

            {

                dis[v]=w+dis[u];

                if(!vis[v])

                {

                    vis[v]=1;

                    cnt[v]++;

                    if(cnt[v]>n)return 1;

                    que.push(v);

                }

            }

        }

    }

    return 0;

}

int main()

{

    int T,u,v,w;

    scanf("%d",&T);

    while(T--)

    {

        scanf("%d%d%d",&n,&m,&s);

        init();

        for(int i=1;i<=m;i++)

        {

            scanf("%d%d%d",&u,&v,&w);

            addedge(u,v,w);

            addedge(v,u,w);

        }

        for(int i=1;i<=s;i++)

        {

            scanf("%d%d%d",&u,&v,&w);

            addedge(u,v,-w);

        }

        if(spfa())puts("YES");

        else puts("NO");

    }

}

View Code