POJ 3268 Silver Cow Party bfs最短路

题目链接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66569#problem/D

题意：cows要举办一个party，地点选择其中一个cow的位置，其他cow去参加，需要沿着单向边走一个来回，求他们的最小来回路程中的最大的值。

刚开始天真的写了N多个DFS，然后各种WA，只好放弃了，然后改写BFS，然后AC，然后就没有然后了，看来最短路还是写成BFS吧、。。。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
#define INF 0x7fffffff

using namespace std;

struct edge{
    int u,v,w;
}E[100010];

int N,M,X;
vector<int> G[2][1010];
int ans[2][1010],vis[1010];

void bfs(int op){//op为0时正向更新v,op为1时反向更新u
    memset(vis,0,sizeof(vis));
    for(int i=1;i<=N;i++) ans[op][i]=INF;
    queue<int> Q;
    Q.push(X);
    vis[X]=1;ans[op][X]=0;
    while(!Q.empty()){
        int tp=Q.front();Q.pop();vis[tp]=0;
        for(int i=0;i<(int)G[op][tp].size();i++){
            int t=G[op][tp][i];
            int u=E[t].u,v=E[t].v,w=E[t].w;
            if(op) swap(u,v);//反向更新最小值
            if(ans[op][u]+w>=ans[op][v]) continue;
            ans[op][v]=ans[op][u]+w;
            if(!vis[v]) vis[v]=1,Q.push(v);
        }
    }
}


int main(){
    freopen("D:\\in.txt","r",stdin);
    while(cin>>N){
        cin>>M>>X;
        int a,b,c;
        for(int i=0;i<=N;i++) G[0][i].clear(),G[1][i].clear();
        for(int i=0;i<M;i++){
            scanf("%d %d %d",&a,&b,&c);
            E[i]=(edge){a,b,c};
            G[0][a].push_back(i);//以a出发的边
            G[1][b].push_back(i);//指向b的边
        }
        bfs(0);bfs(1);
        int Max=0;
        for(int i=1;i<=N;i++) {
            Max=max(Max,ans[0][i]+ans[1][i]);
        }
        cout<<Max<<endl;
    }
    return 0;
}