POJ3268---Silver Cow Party （最短路径）

题目来源：https://vjudge.net/problem/POJ-3268
【题意】
给出n个点，一个点x，求其他点到这个点再返回原来位置的最短路径。
并且，给出的边是单向边。
【思路】
相信从一个定点到其他所有点的最短路径很好求，只需要一个spfa即可，但是呢，题目要求还有一部分，从其他所有点到一个定点的最短路径，额。。有点尴尬，想了好一会，想过用三层for的folyed，但是2s的话会超时的对吧。自行否定，然后就想猜一下，是否把 w[][]二维数组里的值交换一下端点，然后再来一遍以x为定点向其他点的最短路径（spfa）就可以了？，，，然后试了一下，发现可以，，，，就过了。。。
原来还有这种操作。原理就是，单向边的方向变了，那么从定点到其他点就相当于从其他点到定点。。。
【代码】

#include
#include
#include
#include
#include
#include
#define INF 0x3f3f3f3f
using namespace std;
int w[1010][1010];
int d1[1010];
int d2[1010];
bool vis[1010];
int n,m,x;
struct cmp
{
    bool operator() (int &a,int &b)
    {
        return a>b;
    }
};
void init()
{
    for(int i=1; i<=n; i++)
        for(int j=1; j<=n; j++)
            w[i][j]=i==j?0:INF;
    for(int i=1; i<=n; i++)
    {
        d1[i]=i==x?0:INF;
        d2[i]=d1[i];
        vis[i]=0;
    }
}
void spfa()
{
    priority_queue<int,vector<int>,cmp>que;
    que.push(x);
    vis[x]=1;
    while(!que.empty())
    {
        int res=que.top();
        que.pop();
        vis[res]=0;
        for(int i=1; i<=n; i++)
        {
            if(d1[i]>d1[res]+w[res][i])
            {
                d1[i]=d1[res]+w[res][i];
                que.push(i);
                vis[i]=1;
            }
        }
    }
    memset(vis,0,sizeof(vis));
    que.push(x);
    vis[x]=1;
    for(int i=1; i<=n; i++)
        for(int j=i+1; j<=n; j++)
            swap(w[i][j],w[j][i]);
    while(!que.empty())
    {
        int res=que.top();
        que.pop();
        vis[res]=0;
        for(int i=1; i<=n; i++)
        {
            if(d2[i]>d2[res]+w[res][i])
            {
                d2[i]=d2[res]+w[res][i];
                que.push(i);
                vis[i]=1;
            }
        }
    }
    int maxx=-1;
    for(int i=1;i<=n;i++)
    {
        if(d1[i]+d2[i]>maxx)
            maxx=d1[i]+d2[i];
    }
    printf("%d\n",maxx);
}
int main()
{
    while(~scanf("%d%d%d",&n,&m,&x))
    {
        init();
        for(int i=0; iint x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            if(w[x][y]>z)
                w[x][y]=z;
        }
        spfa();
    }
}