POJ 3259 Wormholes floyd

点击打开链接

题意：一个图，有权值为正的双向边和权值为负的单向边，寻找负环。

题解：floyd，检查的时候看mp[i][i]是不是有负的。注意这里要放在第二层循环里面检查检查出来就返回，因为放在第一层不能检查所有的mp[i][i]，放在第三层浪费了计算。三层循环之后，另起一个循环检查就得每次都跑完三层循环，也会超时。

#include<cstdio>
#include<iostream>
#include<cstring>
using namespace std;
int mp[505][505];
int n,m,w,f;
bool floyd(){
    int i,j,k;
    for(k=1;k<=n;k++){
        for(i=1;i<=n;i++){
            for(j=1;j<=n;j++){
                if(mp[i][j]>(mp[i][k]+mp[k][j]))mp[i][j]=mp[i][k]+mp[k][j];
            }
        }
    }
    for(i=1;i<=n;i++)if(mp[i][i]<0)return true;
    return false;
}
int main(){
    int i,j;
    int s,e,t;
    scanf("%d",&f);
    while(f--){
        scanf("%d%d%d",&n,&m,&w);
        for(i=1;i<=n;i++)
            for(j=1;j<=m;j++)mp[i][j]=10005;
        for(i=1;i<=m;i++){
            scanf("%d%d%d",&s,&e,&t);
            mp[s][e]=mp[e][s]=min(t,mp[e][s]);
        }
        for(i=1;i<=w;i++){
            scanf("%d%d%d",&s,&e,&t);
            mp[s][e]=-t;
        }
        bool flag;
        flag=floyd();
        if(flag)printf("YES\n");
        else printf("NO\n");
    }
}