Wormholes 虫洞 HYSBZ - 1715 【SPFA算法】

 

John在他的农场中闲逛时发现了许多虫洞。虫洞可以看作一条十分奇特的有向边，并可以使你返回到过去的一个时刻（相对你进入虫洞之前）。John的每个农场有M条小路（无向边)连接着N （从1..N标号）块地，并有W个虫洞。其中1<=N<=500,1<=M<=2500,1<=W<=200。 现在John想借助这些虫洞来回到过去（出发时刻之前），请你告诉他能办到吗。 John将向你提供F(1<=F<=5)个农场的地图。没有小路会耗费你超过10000秒的时间，当然也没有虫洞回帮你回到超过10000秒以前。

Input

* Line 1: 一个整数 F, 表示农场个数。

* Line 1 of each farm: 三个整数 N, M, W。

* Lines 2..M+1 of each farm: 三个数（S, E, T）。表示在标号为S的地与标号为E的地中间有一条用时T秒的小路。

* Lines M+2..M+W+1 of each farm: 三个数（S, E, T）。表示在标号为S的地与标号为E的地中间有一条可以使John到达T秒前的虫洞。

Output

* Lines 1..F: 如果John能在这个农场实现他的目标，输出"YES",否则输出"NO"。

Sample Input

2 3 3 1 1 2 2 1 3 4 2 3 1 3 1 3 3 2 1 1 2 3 2 3 4 3 1 8

Sample Output

NO YES

 

用spfa算法，套模板，要考虑虫洞，因为虫洞会使农场主返回c秒之前，所以add_edge(a,b,-c);

//spafa
//复杂度O(KE)
#include
#include
#include
#include
#include

using namespace std;

const int MAXN=1010;
const int INF=0x3f3f3f3f;

struct edge{
    int v,cost;
    edge(int _v,int _cost):v(_v),cost(_cost){}
};
vectorE[MAXN];

void add_edge(int u,int v,int w)
{
    E[u].push_back(edge(v,w));
}

bool vis[MAXN];
int cnt[MAXN];
int dist[MAXN];

bool spfa(int start,int n)
{
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;i++){
        dist[i]=INF;
    }
    vis[start]=true;
    dist[start]=0;
    queueque;
    while(!que.empty()){
        que.pop();
    }
    que.push(start);
    memset(cnt,0,sizeof(cnt));
    cnt[start]=1;
    while(!que.empty()){
        int u=que.front();
        que.pop();
        vis[u]=false;
        for(int i=0;idist[u]+E[u][i].cost){
                dist[v]=dist[u]+E[u][i].cost;
                if(!vis[v]){
                    vis[v]=true;
                    que.push(v);
                    if(++cnt[v]>n){
                        return false;
                    }
                }
            }
        }
    }
    return true;
}

int main()
{
    int f;
    scanf("%d",&f);
    while(f--){
        int n,m,w;
        scanf("%d%d%d",&n,&m,&w);
        for(int i=1;i<=n;i++){
            E[i].clear();
        }
        int a,b,c;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&a,&b,&c);
            add_edge(a,b,c);
            add_edge(b,a,c);
        }
        //虫洞，负边
        for(int i=1;i<=w;i++){
            scanf("%d%d%d",&a,&b,&c);
            add_edge(a,b,-c);
        }
        if(spfa(1,n))
            printf("NO\n");
        else
            printf("YES\n");
    }

    return 0;
}