B - 数据结构实验之图论十一：AOE网上的关键路径

题目链接 https://acm.sdut.edu.cn/onlinejudge3/contests/3990/problems/B

#include<bits/stdc++.h>
#define maxn 50010
using namespace std;
int n,m,t;//t源点
int dis[maxn],path[maxn],in[maxn];
//dis(i到汇点的最大路径)，path(存放i的下一个点)，in(入度)

struct node
{
    int u,v,w;
}edge[maxn];

void bellman(int n,int m)
{
    for(int i=2; i<=n; i++)
    {//除去源点，只需要进行n-1次求值
        int flag=0;//标记是否所有点都已更新
        for(int j=1; j<=m; j++)
        {
            if((dis[edge[j].u]<dis[edge[j].v]+edge[j].w)||((dis[edge[j].u]==dis[edge[j].v]+edge[j].w)&&(edge[j].v<path[edge[j].u])))
            {
                dis[edge[j].u]=dis[edge[j].v]+edge[j].w;
                path[edge[j].u]=edge[j].v;//如果有两条路径权值相同，保留字典序较小的
                flag=1;
            }
        }
        if(flag==0) break;
    }
    cout<<dis[t]<<endl;
    int k=t;
    while(path[k]!=0)
    {
        cout<<k<<" "<<path[k]<<endl;
        k=path[k];
    }
}

int main()
{
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        memset(in,0,sizeof(in));
        memset(dis,0,sizeof(dis));
        memset(path,0,sizeof(path));
        for(int i=1; i<=m; i++)
        {
            int u,v,w;
            cin>>u>>v>>w;
            edge[i].u=u;
            edge[i].v=v;
            edge[i].w=w;
            in[v]++;
        }
        for(int i=1; i<=n; i++)
        {//寻找源点
            if(in[i]==0) t=i;
        }
        bellman(n,m);
    }
    return 0;
}