POJ 3259 Wormholes(Floyd判负环)

 

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

       题意是有一个n*n的地图，输入m条无向边，输入k条单向的虫洞（它的权值是负的，因为可以让时间倒流...），然后问能不能从一个点出发转一圈后回到这个点的时间是负的...也就是让时间倒流...

       直接用floyd把每个边都更新一下，然后看有没有pre[i][i](自己到自己)的权值是负的就好了...还可以用spfa去判负环，可以看下这篇博客：POJ 3259 Wormholes(spfa判负环)

---

AC代码：

#include 
#include 
#include 
#define inf 0x3f3f3f3f
#define maxn 505
using namespace std;
int T,n,m,k;
int pre[maxn][maxn];

void init(){
  for(int i=0;i<=n;i++){
    for(int j=0;j<=n;j++){
      if(i == j)pre[i][j] = 0;
      pre[i][j] = pre[j][i] = inf;
    }
  }
}

bool floyd(){
  for(int k=1;k<=n;k++){
    for(int i=1;i<=n;i++){
      for(int j=1;j<=n;j++){
        if(pre[i][j] > pre[i][k] + pre[k][j]){
          pre[i][j] = pre[i][k] + pre[k][j];
        }
      }
      if(pre[i][i] < 0)return true;
    }
  }
  return false;
}

int main()
{
  scanf("%d",&T);
  while(T--){
    scanf("%d%d%d",&n,&m,&k);
    init();
    while(m--){
      int u,v,l;
      scanf("%d%d%d",&u,&v,&l);
      if(l < pre[u][v])
      pre[u][v] = pre[v][u] = l;
    }
    while(k--){
      int u,v,l;
      scanf("%d%d%d",&u,&v,&l);
      pre[u][v] = -l;
    }
    if(floyd()) puts("YES");
    else puts("NO");
  }
  return 0;
}