【最短路】【水题】【cogs 497】奶牛派对

497. 奶牛派对

★☆   输入文件:party.in   输出文件:party.out   简单对比
时间限制:1 s   内存限制:128 MB

【问题描述】

N头牛要去参加一场在编号为x(1≤x≤n)的牛的农场举行的派对(1≤N≤1000),有M(1≤m≤100000)条有向道路,每条路长ti(1≤ti≤100);每头牛都必须参加完派对后回到家,每头牛都会选择最短路径,求这n个牛的最短路径(一个来回)中最长的一条的长度。

特别提醒:可能有权值不同的重边。

输入格式:

第一行:N,M,X;
第二--m+1行:Ai,Bi,Ti,表示有一条从Ai到Bi的路,长度为Ti.

输出格式:

最长最短路的长度。

样例输入输出:
party.in

4 8 2
1 2 4
1 3 2
1 4 7
2 1 1
2 3 5
3 1 2
3 4 4
4 2 3

party.out

10

题解:
既然是从许多点到一个点,我们可以看成是从一个点到许多点,然后跑SPFA即可,因为是来回,所以正着建一遍边,在反着建一遍就好了。

Code:

#include
#include
#include
#include
#include
#include
#define N 1010
#define M 1000100
using namespace std;
struct Edge{
    int v,next,d;
}edge[M];
struct E{
    int u,v,d;
}e[M];
int n,m,S,num=0,ans=0,head[N],dis[2][N],q[M];
bool vis[N];
int in(){
    int x=0; char ch=getchar();
    while (ch<'0' || ch>'9') ch=getchar();
    while (ch>='0' && ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
void add(int u,int v,int d){
    edge[++num].v=v; edge[num].d=d;
    edge[num].next=head[u]; head[u]=num;
}
void spfa(int p){
    int h=0,t=1; int k=0;
    memset(dis[p],0x3f,sizeof(dis[p]));
    memset(vis,0,sizeof(vis));
    dis[p][S]=0; vis[S]=1; q[h]=S;
    while (hint u=q[h++];vis[u]=0;
        for (int i=head[u]; i; i=edge[i].next){
            int v=edge[i].v;
            if (dis[p][v]>dis[p][u]+edge[i].d){
                dis[p][v]=dis[p][u]+edge[i].d;
                if (!vis[v]) vis[v]=1,q[t++]=v;
            }
        }
    }
}
int main(){
    memset(head,0,sizeof(head));
    n=in(),m=in(),S=in();
    for (int i=1; i<=m; i++){
        int u=in(),v=in(),k=in();
        e[i].u=u,e[i].v=v,e[i].d=k;
        add(u,v,k);
    }
    spfa(0);
    num=0; memset(head,0,sizeof(head));
    for (int i=1; i<=m; i++)
        add(e[i].v,e[i].u,e[i].d);
    spfa(1);
    for (int i=1; i<=n; i++)
        ans=max(ans,dis[0][i]+dis[1][i]);
    printf("%d\n",ans);
    return 0;
}

你可能感兴趣的:(算法——最短路,算法——基础)