poj3259 Wormholes (bellman-ford判断负权环是否存在)

Wormholes

poj3259

问题描述

在探索他的许多农场时，农夫约翰发现了许多令人惊奇的虫洞。虫洞是非常特殊的，因为它是一个单向的路径，把你送到它的目的地，在你进入虫洞之前的时间!FJ的每个农场包括N(1≤N≤500)块，方便编号为1。N, M(1≤M≤2500)条路径，W(1≤W≤200)条虫洞。
由于FJ是一个狂热的时间旅行爱好者，他想做以下事情:从某个领域开始，通过一些路径和虫洞旅行，并在他最初出发前回到起始领域。也许他能遇见他自己:)。
为了帮助FJ知道这是否可能，他会提供给你他农场F(1≤F≤5)的完整地图。任何路径的运行时间都不会超过1万秒，也没有虫洞可以让FJ回到超过1万秒的时间。

输入

第1行:一个整数，后面是F - F farm描述。
每个farm的第1行:三个空格分隔的整数:N、M和W
行2 . .每个农场的M+1:三个空格分隔的数字(S, E, T)，分别描述:S和E之间的双向路径，需要T秒的时间。两个字段可能由多个路径连接。
行M + 2 . .每个农场的M+ W+1:三个用空格分隔的数字(S, E, T)分别描述:从S到E的单向路径，也将旅行者向后移动了T秒。

输出

行1 . .F:对于每个农场，如果FJ能够达到他的目标，输出“YES”，否则输出“NO”(不包含引号)。

分析：

bellman-ford判断是否存在负权回路即可，注意虫洞是单向的
(以前一直拖着没学bellmanford和spfa)

我的代码：

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<string>
#include<vector>
#include<set>
#include<queue>
#include<stack>
#include<map>
#include<string>
#include<algorithm>
#include<sstream>
#include<memory>
#include<utility>
#include<functional>
#include<iterator>
typedef long long ll;
const int inf=0x3f3f3f3f;
const int inn=0x80808080;
using namespace std;
const int maxm=1e5;
int d[maxm];
int u[maxm];
int v[maxm];
int w[maxm];
int n,m,k;
void bellman(){
    memset(d,inf,sizeof d);
    d[1]=0;
    for(int jj=1;jj<=n-1;jj++){
        int mark=0;//判断是否改变
        for(int i=1;i<=m;i++){
            if(d[u[i]]>d[v[i]]+w[i]){
                d[u[i]]=d[v[i]]+w[i];
                mark=1;
            }
            if(d[v[i]]>d[u[i]]+w[i]){
                d[v[i]]=d[u[i]]+w[i];
                mark=1;
            }
        }
        for(int i=m+1;i<=m+k;i++){
            if(d[v[i]]>d[u[i]]+w[i]){
                d[v[i]]=d[u[i]]+w[i];
                mark=1;
            }
        }
        if(mark==0)break;//没改说明已经到达最短了
    }
    for(int i=1;i<=m;i++){
        if(d[u[i]]>d[v[i]]+w[i]){
            printf("YES\n");
            return ;
        }
    }
    printf("NO\n");
}
int main(){
    int T;
    cin>>T;
    while(T--){
        cin>>n>>m>>k;
        for(int i=1;i<=m;i++){
            cin>>u[i]>>v[i]>>w[i];
        }
        for(int i=m+1;i<=m+k;i++){
            cin>>v[i]>>u[i]>>w[i];//虫洞u->v是正权，这里改成v->u负权
            w[i]=-w[i];
        }
        bellman();
    }
    return 0;
}