POJ 3259 Wormholes

最短路问题。这个题也够DT。。出题者真会YY。

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

两个算法。一个是bellman-ford一个就是他的队列实现。也就是对他的优化spfa算法。

bellman-ford算法

#include<stdio.h>
#include<string.h>
#define INF 1000000
int N,F,M,W,en;//N 农场个数，F 农田个数 M路径条数 W虫洞个数 en边数
int dist[510];
typedef struct node
{
    int u,v;//边的两个端点
    int w;//权值
}EDGE;
EDGE e[INF];
void bellman()
{
    int i,j,flag;
    for(i=0;i<N-1;i++)
    {
        flag=0;
        for(j=0;j<en;j++)
    {
    if (dist[e[j].v] > dist[e[j].u]+e[j].w)
    {
    flag=1;
    dist[e[j].v] = dist[e[j].u]+e[j].w;
    }
    }
    }
    for(j=0;j<en;j++)
    {
     if(dist[e[j].v]>dist[e[j].u]+e[j].w)
     {
         printf("YES\n");
         return;
     }
    }
    printf("NO\n");

}
int main()
{
    int i;
    scanf("%d",&F);
    while(F--)
    {
        memset(dist,INF,sizeof(dist));
        en=0;
        scanf("%d%d%d",&N,&M,&W);
        for(i=0;i<M;i++)
        {
            scanf("%d%d%d",&e[en].u,&e[en].v,&e[en].w);
            en++;
            e[en].u=e[en-1].v;
            e[en].v=e[en-1].u;
            e[en].w=e[en-1].w;
            en++;
        }
       for (i=0;i<W;i++)
     {
      scanf("%d%d%d",&e[en].u,&e[en].v,&e[en].w); //虫洞路径  u， v， 以及倒流的时间
      e[en].w=-e[en].w;
       en++;
     }
     bellman();
    }
    return 0;
}

spfa算法

#include<iostream>
#include<stdio.h>
#include<algorithm>
#include<math.h>
#include<queue>
#include<string.h>
#define Max 0xfffffff
using namespace std;
const int N=505;
int path[505][505];//边距
int dist[505];//到原点距离
bool spfa(int n)
{
    queue<int>str;
    bool in[N];
    int num[N];
    memset(in,false,sizeof(in));//是否在队列中
    memset(num,0,sizeof(num));//进队列次数
    for(int i=1;i<=n;++i)//初始化
    dist[i]=Max;
    dist[1]=0;
    str.push(1);
    in[1]=true;
    num[1]=1;
    while(!str.empty())//队列是否为空 如果队列一直到空 则说明没有负环
    {
        int x=str.front();
        str.pop();
        in[x]=false;
        for(int i=1;i<=n;++i)
        {
            if(dist[i]>dist[x]+path[x][i])//找可更新点
            {
                dist[i]=dist[x]+path[x][i];
                if(in[i]==false)//如果可更新点不再队列中
                {
                    str.push(i);//进队列
                    in[i]=true;//标记
                    num[i]++;//进队列次数
                    if(num[i]==n)//进队列此次到达n  说明有负环
                    return true;
                }
            }
        }
    }
    return false;
}
int main()
{
   // freopen("data.txt","r",stdin);
    int n,m,w;
    int i,j,l;
    int f;
    cin>>f;
    while(f--)
    {
        cin>>n>>m>>w;
         for(i=1;i<=n;i++)
        {
            for(j=i;j<=n;j++)
            {
                path[i][j]=path[j][i]=Max;
            }
        }
        while(m--)
        {
           cin>>i>>j>>l;
           if(l<path[i][j])
           {
               path[i][j]=path[j][i]=l;
           }
        }
        while(w--)
        {
            cin>>i>>j>>l;
            if((-l)<path[i][j])
            {
                path[i][j]=(-l);
            }
        }
        if(spfa(n)==true)
        printf("YES\n");
        else
        printf("NO\n");
    }
    return 0;
}