虫洞（bellman）

描述

地球存在许多令人惊叹的虫洞。虫洞非常奇特，它是一条单向路径，可以在进入虫洞之前的某个时间内将你送到目的地！一共有N（1≤N≤500）个编号为1...N的区域、M条路径和W个虫洞

从某个区域开始，你通过一些路径和虫洞，在他出发前一段时间回到起始场地。也许你能见到自己：）

为你提供F（1≤F≤5）个完整地图。没有任何路径的通过时间超过10000秒，也没有任何虫洞能让你在时间上倒退超过10000秒

输入描述

第一行：一个整数 F ，描述如上。
接下来对每个农场，第一行三个整数，空格分隔：N、M、W。
接下来 M 行，每行三个整数，空格分隔：S、E、T，分别描述：存在点 S 到点 E 的双向路径，单程花费 T 秒。两块区域之前可以被超过一条路连接。
接下来 W 行，每行三个整数，空格分隔：S、E、T，分别描述：存在点 S 到点 E 的单向路径，能够令进入者回到 T 秒之前。

输出描述

一共1…F行: 对每个地图，如果能成功, 则输出"YES" 否则输出 “NO”

用例输入 1 

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

用例输出 1 

NO
YES

提示

对地图 1，你并不能穿越到出发之前。
对地图2，你可以沿闭环 1→2→3→1 行进，在他出发之前回到出发地 1。他可以从这个环路上的任意位置出发来完成目标。

来源

bellman

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int N = 505,M = 5205;
int idx,h[N],e[M],w[M],ne[M];
int n,m1,m2,d[N],cnt[N],q[N];
bool st[N];
void add(int a,int b,int c){
    e[idx] = b,w[idx] = c,ne[idx] = h[a],h[a] = idx++;
}
bool spfa(){
    memset(st,false,sizeof st);
    memset(d,0,sizeof d);
    memset(cnt,0,sizeof cnt);
    int hh = 0,tt = 0;
    for(int i = 1;i <= n;i++){
        q[tt++] = i;
        st[i] = true;
    }
    while(hh != tt){
        int u = q[hh++];
        if(hh == N) hh = 0;
        st[u] = false;
        for(int i = h[u];~i;i = ne[i]){
            int j = e[i];
            if(d[u] + w[i] < d[j]){
                d[j] = d[u] + w[i];
                cnt[j] = cnt[u] + 1;
                if(cnt[j] >= n) return true;
                if(!st[j]){
                    q[tt++] = j;
                    if(tt == N) tt = 0;
                    st[j] = true;
                }
            }
        }
    }
    return false;
}
int main()
{
    int T;
    cin>>T;
    while(T--){
        cin>>n>>m1>>m2;
        int a,b,c;
        memset(h,-1,sizeof h);
        idx = 0;
        for(int i = 0;i < m1;i++){
            cin>>a>>b>>c;
            add(a,b,c),add(b,a,c);
        }
        for(int i = 0;i < m2;i++){
            cin>>a>>b>>c;
            add(a,b,-c);
        }
        if(spfa())  puts("YES");
        else    puts("NO");
    }
    return 0;
}