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

求负权图的单源最短路

代码在这：

#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;
}