Silver Cow Party(dijkstra)

原题链接原题
做的第一道最短路 明显比模板题要难一点点。。。。。
求牛来回的最短路中的最长时间 路是单向的,所以你可以用一个way数组记录一下来时的时间
返回的时候你如果求各个农场到目标农场的最短路就太耗时了 所以你可以反转一下来和去的路
下面是代码

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int map[1002][1002],dis[1002],way[1002];
int n,x,inf=999999;
void dijkstra()
{
    int book[1002],i,j,u;
    memset(book,0,sizeof(book));
    for(i=1;i<=n;i++)
        dis[i]=map[x][i];
    book[x]=1;
    for(i=1;i<n;i++)
    {
        int min=inf;
        for(j=1;j<=n;j++)
        {
            if(!book[j]&&dis[j]<min)
            {
                u=j;
                min=dis[j];
            }
        }
        book[u]=1;
        for(j=1;j<=n;j++)
        {
            if(!book[j]&&map[u][j]+dis[u]<dis[j])
                dis[j]=map[u][j]+dis[u];
        }
    }
}

int main()
{
    int m,i,j,a,b,t;
    while(~scanf("%d%d%d",&n,&m,&x))
    {
        for(i=1;i<=n;i++)
            for(j=1;j<=n;j++)
        {
            if(i==j)
                map[i][j]=0;
            else
                map[i][j]=inf;
        }
        while(m--)
        {
            scanf("%d%d%d",&a,&b,&t);
            if(map[a][b]>t)
                map[a][b]=t;
        }
        dijkstra();
        for(i=1;i<=n;i++)
            way[i]=dis[i];
        for(i=1;i<=n;i++)
            for(j=i+1;j<=n;j++)
            swap(map[i][j],map[j][i]);//反转
        dijkstra();
        int ans=0;
        for(i=1;i<=n;i++)
        {
            if(i!=x)
                ans=max(ans,way[i]+dis[i]);
        }
        printf("%d\n",ans);
    }
    return 0;
}
发布了17 篇原创文章 · 获赞 3 · 访问量 521

猜你喜欢

转载自blog.csdn.net/RUBGH/article/details/104855975