POJ 3259 Wormholes （最短路BF）

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

题目意思

在一个农场出现了w个虫洞，经过虫洞会从u到v并且回到t秒前。场主给了m条路径和所花费的时间。现在农场主想知道在农村是否存在一个从v到u，然后经过虫洞从u回到v看到曾今的自己。

解题思路

最短路判断是否存在负环，把虫洞的时间改回负的跑一遍BF看是否存在负环就可以了。（从哪个点开始跑并不影响）

代码部分

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <string.h>
#include <queue>
#include <string>
#include <map>
using namespace std;
#define LL long long
#define inf 0x3f3f3f3
const int N = 1e4;
double dis[N];
int n,m,w;
struct node
{
    int v,u,t;
}e[N];
bool BF()
{
    for (int i = 0 ;i < n; i++)
    {
        dis[i] = 999999;
    }
    dis[1] = 0;
    for (int i = 1; i < n; i++)
    {
        bool fa = false;
        for (int j = 0; j < 2*m+w; j++)
        {
            int v = e[j].v;
            int u = e[j].u;
            if (dis[v] > dis[u] + e[j].t)
            {
                fa = true;
                dis[v] = dis[u] + e[j].t;
            }
        } 
        if (!fa)
            return false;
    }
    for (int j = 0; j < 2*m+w; j++)
    {
        int v = e[j].v;
        int u = e[j].u;
        if (dis[v] > dis[u] + e[j].t)
        {
            return true;
        }
    }
    return false;
}
int main()
{
    int T;
    scanf("%d",&T);
    while (T--)
    {
        int u,v,t;
        scanf("%d %d %d",&n,&m,&w);
        for (int i = 0;i < m; i++)
        {
            scanf("%d %d %d",&e[i].u,&e[i].v,&e[i].t);  ///双向存边 
            e[m+i].u = e[i].v;
            e[m+i].v = e[i].u;
            e[m+i].t = e[i].t;  
        }
        for (int i = 0; i < w; i++)
        {
            scanf("%d %d %d",&e[2*m+i].u,&e[2*m+i].v,&e[2*m+i].t); ///存储虫洞 
            e[2*m+i].t = -e[2*m+i].t;   ///时间改为减 
        }
        if (BF())
            cout<<"YES"<<endl;
        else cout<<"NO"<<endl;
    } 
    return 0;
}