POJ 3259（最短路径）

题目链接：http://poj.org/problem?id=3259

题意：现在有n个点，m条边，w个虫洞，给出m个a,b,c，表示从a到b需要花费c时间，从b到a需要话费c时间，然后给出w个a,b,c，表示从a到b花费-c个时间，问是否存在一条路径使得最终回到源点的时间比初始时间小。

思路：最短路径，Bellman-Ford或SPFA判断是否存在负环的问题。

注意：数据范围比题目中给的略大一点，这点略坑。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;

const int INF=0x3f3f3f3f;
const int maxn=1010;
int T,n,m,w;
int dis[maxn];
int eNum;

struct Edge{
int s;
int e;
int t;
}edge[10000];

void addEdge(int s,int e,int t){
edge[eNum].s=s;
edge[eNum].e=e;
edge[eNum].t=t;
eNum++;
}

bool Bellman_Ford(){
for(int i=1;i<=n;i++)
dis[i]=INF;
dis[1]=0;
for(int i=1;i<n;i++){
for(int j=0;j<eNum;j++){
if(dis[edge[j].e]>dis[edge[j].s]+edge[j].t){
dis[edge[j].e]=dis[edge[j].s]+edge[j].t;
}
}
}
for(int k=0;k<eNum;k++){
if(dis[edge[k].e]>dis[edge[k].s]+edge[k].t){
return true;
}
}
return false;
}

int main(){
#ifndef ONLINE_JUDGE
freopen("test.in","r",stdin);
freopen("test.out","w",stdout);
#endif
scanf("%d",&T);
while(T--){
scanf("%d%d%d",&n,&m,&w);
eNum=0;
int s,e,t;
for(int i=0;i<m;i++){
scanf("%d%d%d",&s,&e,&t);
addEdge(s,e,t);
addEdge(e,s,t);
}
for(int i=0;i<w;i++){
scanf("%d%d%d",&s,&e,&t);
addEdge(s,e,-t);
}
bool ans=Bellman_Ford();
if(ans) puts("YES");
else puts("NO");
}
return 0;
}