POJ3259----Wormholes(SPFA判断负环)

 判断给定的有向图中是否存在负环。

      利用 spfa 算法判断负环有两种方法：

      1） spfa 的 dfs 形式，判断条件是存在一点在一条路径上出现多次。

      2） spfa 的 bfs 形式，判断条件是存在一点入队次数大于总顶点数。

DFS

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<math.h>
#include<queue>
#include<vector>
#include<string>
#include<string.h>
#include<algorithm>
using namespace std;
#define inf 0xffffff
int n,m,w,flag;
struct e{int des,val;};
int vis[520],dis[520];
vector<e>v[520];
void spfa(int k)
{
if(flag) return ;
vis[k]=1;
int len=v[k].size();
for(int i=0;i<len;i++){
int now=v[k][i].des;
if(dis[now]>dis[k]+v[k][i].val){
dis[now]=dis[k]+v[k][i].val;
if(vis[now]&&!flag){
printf("YES\n");
flag=1;
}
else{
vis[now]=1;
spfa(now);
}
}
}
vis[k]=0;
}
void init()
{
memset(vis,0,sizeof(vis));
for(int i=2;i<=n;i++) dis[i]=inf;
dis[1]=0;
flag=0;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&w);
for(int i=1;i<=n;i++)
v[i].clear();

int st,ed,va;
e a;
for(int i=0;i<m;i++){
scanf("%d%d%d",&st,&ed,&va);
a.des=ed,a.val=va;
v[st].push_back(a);

a.des=st,a.val=va;
v[ed].push_back(a);
}
for(int i=0;i<w;i++){
scanf("%d%d%d",&st,&ed,&va);
a.des=ed,a.val=-va;
v[st].push_back(a);
}
init();
spfa(1);
if(!flag) printf("NO\n");
}
}

BFS

#include<stdio.h>
#include<algorithm>
#include<iostream>
#include<math.h>
#include<queue>
#include<vector>
#include<string>
#include<string.h>
#include<algorithm>
using namespace std;
#define inf 0xffffff
int n,m,w;
struct e{int des,val;};
vector<e>v[520];
int spfa()
{
int vis[520],dis[520],cnt[520];
for(int i=0;i<=n;i++){
vis[i]=0,dis[i]=inf,cnt[i]=0;
}
dis[1]=0,vis[1]=1,cnt[1]=1;

queue<int> q;
q.push(1);
while(!q.empty()){
int u=q.front();
q.pop();
vis[u]=0;
int len=v[u].size();
for(int i=0;i<len;i++)
{
int de=v[u][i].des;
int va=v[u][i].val;
if(dis[de]>dis[u]+va)
{
dis[de]=dis[u]+va;
if(!vis[de])
{
vis[de]=1;
cnt[de]++;
if(cnt[de]>=n) return 1;
q.push(de);
}
}
}
}
return 0;
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d%d",&n,&m,&w);
for(int i=1;i<=n;i++)
v[i].clear();
int st,ed,va;
e a;
for(int i=0;i<m;i++){
scanf("%d%d%d",&st,&ed,&va);
a.des=ed,a.val=va;
v[st].push_back(a);

a.des=st,a.val=va;
v[ed].push_back(a);
}
for(int i=0;i<w;i++){
scanf("%d%d%d",&st,&ed,&va);
a.des=ed,a.val=-va;
v[st].push_back(a);
}
int ans=spfa();
if(ans==1) printf("YES\n");
else printf("NO\n");
}
}