Wormholes(POJ 3259)

~题目链接~

http://poj.org/problem?id=3259

Sample Input

2
3 3 1
1 2 2
1 3 4
2 3 1
3 1 3
3 2 1
1 2 3
2 3 4
3 1 8

Sample Output

NO
YES

题目：输入农田个数，连通的路数，虫洞的个数。两农田之间连通是双向的，而虫洞是单向的。虫洞有这样的性质：可是时间倒流。问你这个农民能不能看到他自己，也就是说，有没有这样一条路径，能利用虫洞的时间倒流的性质，让这个人能在这个点出发前回去，这样他就是能看到他自己了。

1.Bellmen_ford

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#define maxn 0x7ffffff

struct node
{
    int u,v,w;
} map[10010];
int h,num[1010];

void cap(int u,int v,int w)
{
    h++;
    map[h].u=u;
    map[h].v=v;
    map[h].w=w;
}

int bellmen_ford(int n)
{
    int i,j;
    for(i=1; i<=n; i++)//初始化
        num[i]=maxn;
    for(i=0; i<n; i++)//迭代n次
        for(j=1; j<=h; j++)//检查每一条边
            if(num[map[j].v]>num[map[j].u]+map[j].w)//松弛
                num[map[j].v]=num[map[j].u]+map[j].w;
    for(i=1; i<=h; i++)//检查是否存在负权回路
        if(num[map[i].v]>num[map[i].u]+map[i].w)
            return 1;
    return 0;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m,k,u,v,w,i;
        h=0;
        scanf("%d%d%d",&n,&m,&k);
        for(i=0; i<m; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            cap(u,v,w);
            cap(v,u,w);
        }
        for(i=0; i<k; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            cap(u,v,-w);
        }
        if(bellmen_ford(n))
            printf("YES
");
        else
            printf("NO
");
    }

    return 0;
}

2.Spfa

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
#include<queue>
#define maxn 0x7ffffff
#define maxm 1000+10

using namespace std;

int map[maxm][maxm],dis[maxm];
queue<int>Q;

int spfa(int n)
{
    while(!Q.empty())
        Q.pop();
    int num[maxm],flag[maxm],i;
    memset(num,0,sizeof(num));
    memset(flag,0,sizeof(flag));
    for(i=1; i<=n; i++)
        dis[i]=maxn;
    Q.push(1);
    dis[1]=0;
    num[1]=1;
    flag[1]=1;
    while(!Q.empty())
    {
        int h=Q.front();
        Q.pop();
        flag[h]=0;
        for(i=1; i<=n; i++)
            if(dis[i]>dis[h]+map[h][i])
            {
                dis[i]=dis[h]+map[h][i];//松弛
                if(!flag[i])
                {
                    Q.push(i);
                    flag[i]=1;
                    num[i]++;
                    if(num[i]>=n)//判断负权回路只需要入队次数大于顶点数就行了
                        return 1;
                }
            }
    }
    return 0;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m,k,u,v,w,i;
        scanf("%d%d%d",&n,&m,&k);
        for(i=0; i<m; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            map[u][v]=map[v][u]=w;
        }
        for(i=0; i<k; i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            map[u][v]=-w;
        }
        if(spfa(n))
            printf("YES
");
        else
            printf("NO
");
    }

    return 0;
}