poj 3259 Wormholes(最短路Bellman-Ford算法)

最新推荐文章于 2022-05-19 19:50:47 发布

原创最新推荐文章于 2022-05-19 19:50:47 发布 · 399 阅读

0 ·

CC 4.0 BY-SA版权

求负权图的单源最短路

代码在这：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

struct node
{
    int x,y,d;
}a[20010];
int dis[50010];
int n,m,w;

int bellmanford(int n)
{
    int x,y,d;
    for(int i=1;i<=n;++i) dis[i]=1e8;
    dis[1]=0;
    for(int i=1;i<n;++i)
    {
        for(int j=1;j<=2*m+w;++j)
        {
            x=a[j].x;
            y=a[j].y;
            d=a[j].d;
            if(dis[x]<1e8&&dis[x]+d<dis[y]) dis[y]=dis[x]+d;
        }
    }
    for(int j=1;j<=2*m+w;++j)
    {
        x=a[j].x;
        y=a[j].y;
        d=a[j].d;
        if(dis[x]<1e8&&dis[x]+d<dis[y]) return 0;
    }
    return 1;
}

int main()
{
    int t;

    scanf("%d",&t);

    while(t--)
    {
        scanf("%d%d%d",&n,&m,&w);

        for(int i=1;i<=m;++i){
            scanf("%d%d%d",&a[2*i-1].x,&a[2*i-1].y,&a[2*i-1].d);
            a[2*i].x=a[2*i-1].y;
            a[2*i].y=a[2*i-1].x;
            a[2*i].d=a[2*i-1].d;
        }
        for(int i=2*m+1;i<=2*m+w;++i){
            scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].d);
            a[i].d=-a[i].d;
        }
        if(!bellmanford(n)) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}