Wormholes

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

Bellman_ford : 判断是否有负权回路  282MS

#include<iostream>
using namespace std ;
#define INF 1>>28
#define N 6000
struct node
{
    int a ;
    int b ;
    int c ;
}e[N] ;
int n, m, w, num ;
int bellman_ford()
{
    int dis[N] ;
    for(int i=1; i<=n; i++)
    dis[i] = INF ;
    dis[1] = 0 ;//以这个为源点
    for(int i=1; i<n; i++)//松弛n-1次就必定找出最短路径
    {
        for(int j=1; j<num; j++)
        if(dis[e[j].b]>dis[e[j].a]+e[j].c)
        dis[e[j].b] = dis[e[j].a] + e[j].c ;
    }
    for(int j=1; j<num; j++)//判断有无负环
    if(dis[e[j].b]>dis[e[j].a]+e[j].c)
    return 1 ;//表示有负环
    return 0 ;
}
int main()
{
    int t ;
    cin>>t ;
    while(t--)
    {
       int a, b, c ;
       num = 1 ;
       cin>>n>>m>>w ;
       for(int i=1; i<=m; i++) //由于paths的双向性，两个方向权值相等，注意指针的移动 
       {
           cin>>a>>b>>c ;
           e[num].a = a ;
           e[num].b = b ;
           e[num].c = c ;
           num++ ;
           e[num].b = a ;
           e[num].a = b ;
           e[num].c = c ;
           num++ ;
       }
       for(int i=1; i<=w; i++)//单向边负环
       {
           cin>>a>>b>>c ;
           e[num].a = a ;
           e[num].b = b ;
           e[num].c = (-1)*c ;
           num++ ;
       }
       if(bellman_ford())
       cout<<"YES"<<endl ;
       else
       cout<<"NO"<<endl ;
    }
    return 0 ;
}

对Bellman_ford的小优化 ,加上flag标志  250MS

#include<iostream>
using namespace std ;
#define INF 1>>28
#define N 6000
struct node
{
    int a ;
    int b ;
    int c ;
}e[N] ;
int n, m, w, num ;
int bellman_ford()
{
    int dis[N] ;
    for(int i=1; i<=n; i++)
    dis[i] = INF ;
    dis[1] = 0 ;//以这个为源点
    for(int i=1; i<n; i++)//松弛n-1次就必定找出最短路径
    {
        bool flag = false ;
        for(int j=1; j<num; j++)
        if(dis[e[j].b]>dis[e[j].a]+e[j].c)
        {
            dis[e[j].b] = dis[e[j].a] + e[j].c ;
            flag = true ; //relax对路径有更新 
        }
        if(!flag)//只要某一次relax没有更新，说明最短路径已经查找完毕，或者部分点不可达，可以跳出relax  
        break ;
    }
    for(int j=1; j<num; j++)//判断有无负环
    if(dis[e[j].b]>dis[e[j].a]+e[j].c)
    return 1 ;//表示有负环
    return 0 ;
}
int main()
{
    int t ;
    cin>>t ;
    while(t--)
    {
       int a, b, c ;
       num = 1 ;
       cin>>n>>m>>w ;
       for(int i=1; i<=m; i++) //由于paths的双向性，两个方向权值相等，注意指针的移动
       {
           cin>>a>>b>>c ;
           e[num].a = a ;
           e[num].b = b ;
           e[num].c = c ;
           num++ ;
           e[num].b = a ;
           e[num].a = b ;
           e[num].c = c ;
           num++ ;
       }
       for(int i=1; i<=w; i++)//单向边负环
       {
           cin>>a>>b>>c ;
           e[num].a = a ;
           e[num].b = b ;
           e[num].c = (-1)*c ;
           num++ ;
       }
       if(bellman_ford())
       cout<<"YES"<<endl ;
       else
       cout<<"NO"<<endl ;
    }
    return 0 ;
}

spfa看是否有一个点进入队列等于n次，375 MS

应该是数据比较少吧spfa竟然最慢

#include <iostream>
#include<cstring>
#include<queue>
#define INF 0x3f3f3f
using namespace std;
struct node
{
    int v,w,next;
}e[6000];
int head[501],vis[501],dis[501],c[501];
int t ;
void init()
{
     t = 0;
     memset(head,-1,sizeof(head));
}
void add(int u,int v,int w)
{
     e[t].v = v;
     e[t].w = w;
     e[t].next = head[u];
     head[u] = t;
     t++;
}
void spfa(int n)
{
     int i, k, flag = 1;
     memset(vis,0,sizeof(vis));
     memset(c,0,sizeof(c));
     queue<int>q;
     for(i = 1; i <= n ; i++)
     dis[i] = INF;
     dis[1] = 0;
     q.push(1);
     vis[1] = 1;
     c[1] = 1;
     while(!q.empty())
     {
         k = q.front();
         if(c[k]==n)
         {
             flag = 0;
             break;
         }

         q.pop();
         vis[k] = 0;
         for(i = head[k];i!=-1 ; i = e[i].next)
         {
             if(dis[k]+e[i].w<dis[e[i].v])
             {
                 dis[e[i].v]=dis[k]+e[i].w;
                 if(!vis[e[i].v])
                 {
                     vis[e[i].v] = 1;
                     q.push(e[i].v);
                     c[e[i].v]++;
                }
             }
         }
     }
     if(!flag)
     cout<<"YES"<<endl ;
     else
     cout<<"NO"<<endl ;
}
 int main()
 {
     int i,n,m,a,b,c,w,p;
     cin>>p ;
     while(p--)
     {
         cin>>n>>m>>w ;
         init();
         for(i = 1; i <= m ; i++)
         {
             cin>>a>>b>>c ;
             add(a,b,c);
             add(b,a,c);
         }
         for(i = 1; i <= w ; i++)
         {
             cin>>a>>b>>c ;
             add(a,b,-c);
        }
         spfa(n);
    }
     return 0;
}