POJ - 3268 Silver Cow Party （Dijkstras算法）

题目链接：

POJ - 3268

题意：

输入三个数据N,M,X。分别代表牛的编号，输入的数据数，牛聚会的地点。现在所有的牛要去X聚会，并且聚会完毕后要回到各自的初始位置，问：所有牛移动的最短距离的最大值是多少。

分析：

这一道题是单向图，将所有的边加入到数组里面后，可以算出来X到各个点的距离，那么接下来如果说我们在算出来各个点到X的距离之后，把这两个距离加起来，然后进行比较就能得出结果。

#include<cstdio>
#include<iostream>
using namespace std;
const int maxn=1e3+10;
const int inf=0x3f3f3f3f;
int N,M,X;
long long int dis[maxn],indexx[maxn],dist[maxn];//距离，标记
long long int mapp[1005][1005];
void Dijkstra(int beginn, int n)//顶点数
{
    memset(indexx,0,sizeof(indexx));
    for(int i=1;i<=n;++i)
    {
        dis[i]=mapp[beginn][i];
    }
    indexx[beginn]=1;//走过
    dis[beginn]=0;//距离为0
    for(int i=1;i<=n;++i)
    {
        long long int minn=inf,v=0;
        for(int j=1;j<=n;++j)
        {
            if(!indexx[j]&&mapp[i][j]<minn)
            {
                minn=mapp[i][j];
                v=j;
            }
        }
        if(!v) break;
        indexx[v]=1;
        for(int j=1;j<=n;++j)
        {
           if(!indexx[j]&&dis[j]>dis[v]+mapp[v][j])
           {
               dis[j]= dis[v]+mapp[v][j];
           }
        }
    }
}
void Dijkstra2(int beginn, int n)//顶点数
{
    memset(indexx,0,sizeof(indexx));
    for(int i=1;i<=n;++i)
    {
        dist[i]=mapp[beginn][i];
    }
    indexx[beginn]=1;//走过
    dist[beginn]=0;//距离为0
    for(int i=1;i<=n;++i)
    {
        long long int minn=inf,v=0;
        for(int j=1;j<=n;++j)
        {
            if(!indexx[j]&&mapp[i][j]<minn)
            {
                minn=mapp[i][j];
                v=j;
            }
        }
        if(!v) break;
        indexx[v]=1;
        for(int j=1;j<=n;++j)
        {
           if(!indexx[j]&&dist[j]>dist[v]+mapp[v][j])
           {
               dist[j]= dist[v]+mapp[v][j];
           }
        }
    }
}
int main()
{
    cin>>N>>M>>X;
    for(int i=1;i<=N;++i)
        for(int j=1;j<=N;++j)
        {
            if(i!=j)
           mapp[i][j]=inf;
           else
            mapp[i][j]=0;
        }
    for(int i=1;i<=M;++i)
    {
        int u,v,w;
        cin>>u>>v>>w;
        mapp[u][v]=w;
    }
    Dijkstra(X,N);
    //现在找到2到各个点的距离，接下来就是找其他点到X的距离
    //X到各个点的距离为dis【i】
    //用dist存储各个点到2的距离
    //并且num_puts存储数据的最大值
    long long int num_puts=-1;
    for(int i=1;i<=N;++i)
    {
        Dijkstra2(i,N);
        if(dist[X]!=inf&&dis[i]!=inf)
        num_puts=max(num_puts,dis[i]+dist[X]);
    }
    printf("%d\n",num_puts);
    return 0;
}

不过，这样做会时间超限。

那再分析一下题意，以题目中给出的例子为例，我们要计算的是从1/3/4到2，再从2到1/3/4的距离，并且我们知道，可以直接使用一个Dijkstra算法求得2到其他各个点的距离，那么如果说现在让结点的顺序改变一下，求得的从2到其他各点的值是不是就是其他各点到2的值呢？

#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
#include<vector>
#include<string>
#include<iostream>
using namespace std;
const int maxn=1e3+10;
const int inf=0x3f3f3f3f;
int N,M,X;
int dis[maxn],indexx[maxn],dist[maxn];
int mapp[maxn][maxn];

void Dijkstra(int beginn, int n)//顶点数
{
    for(int i=1; i<=n; ++i)
    {
        dis[i]=mapp[beginn][i];
        indexx[i]=0;
    }
    indexx[beginn]=1;
    dis[beginn]=0;
    for(int i=1; i<=n; ++i)
    {
        int minn=inf,v=-1;
        for(int j=1; j<=n; ++j)
            if(!indexx[j]&&dis[j]<=minn)
                minn=dis[v=j];
        indexx[v]=1;
        if(v==-1) continue;
        for(int j=1; j<=n; ++j)
        {
            if(dis[j]>dis[v]+mapp[v][j])
            {
                dis[j]= dis[v]+mapp[v][j];
            }
        }
    }
}

void change()
{
    for(int i=1; i<=N; ++i)
    {
        for(int j=1; j<=i; ++j)
        {
               swap(mapp[i][j],mapp[j][i]);
        }
    }
}

int main()
{
    scanf("%d%d%d",&N,&M,&X);
    for(int i=1; i<=N; ++i)
        for(int j=1; j<=N; ++j)
        {
            if(i!=j)
                mapp[i][j]=inf;
            else
                mapp[i][j]=0;
        }
    for(int i=1; i<=M; ++i)
    {
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        mapp[u][v]=w;
    }
    Dijkstra(X,N);
    for(int i=1; i<=N; ++i)
        dist[i]=dis[i];
    change();
    Dijkstra(X,N);
    int num_puts=-1;
    for(int i=1; i<=N; ++i)
    {
        num_puts=max(num_puts,dis[i]+dist[i]);
    }
    printf("%d\n",num_puts);
    return 0;
}

注意点：

1、Dijkstra算法在计算的时候范围是点的数量

2、在Dijkstra算法函数的双层for循环中，第一层循环n-1次就行了

3、

    for(int i=1; i<n; ++i)
    {
        int minn=inf,v=-1;
        for(int j=1; j<=n; ++j)
            if(!indexx[j]&&dis[j]<=minn)
                minn=dis[v=j];
        indexx[v]=1;
        if(v==-1)
            continue;
        for(int j=1; j<=n; ++j)
        {
            if(dis[j]>dis[v]+mapp[v][j])
            {
                dis[j]= dis[v]+mapp[v][j];
            }
        }
    }

这个里面判断的是dis[j]<=minn，而不是mapp[i][j]<=minn

4、初始化！