1003 Emergency (25 分)难度: 中等 / 知识点: 变种的Dijkstra

Posted 2021-09-22 辉小歌

https://pintia.cn/problem-sets/994805342720868352/problems/994805523835109376

#include<bits/stdc++.h>
using namespace std;
const int N=1e3+10;
int dist[N],cnt[N],sum[N],st[N],n,m;
int g[N][N],w[N],startx,endx;
// cnt[j] 存的是j到起点的最短路的数量。
// sum[j] 存的是j到起点的最短路上最大的人数
void Dijkstra()
{
    memset(dist,0x3f,sizeof dist);
    dist[startx]=0,cnt[startx]=1,sum[startx]=w[startx];
    for(int i=0;i<n;i++)
    {
        int t=-1;
        for(int j=0;j<n;j++) if(!st[j]&&(t==-1|| dist[j]<dist[t])) t=j;
        st[t]=1;
        for(int j=0;j<n;j++)
        {
            if(dist[j]>dist[t]+g[t][j])
            {
                dist[j]=dist[t]+g[t][j];
                cnt[j]=cnt[t];
                sum[j]=sum[t]+w[j];
            }
            else if(dist[j]==dist[t]+g[t][j])//相等了
            {
                cnt[j]+=cnt[t];
                sum[j]=max(sum[j],sum[t]+w[j]);
            }
        }
    }
}
int main(void)
{
    memset(g,0x3f,sizeof g);
    cin>>n>>m>>startx>>endx;
    for(int i=0;i<n;i++) cin>>w[i];
    while(m--)
    {
        int a,b,c; cin>>a>>b>>c;
        g[a][b]=g[b][a]=min(g[a][b],c);
    }
    Dijkstra();
    cout<<cnt[endx]<<" "<<sum[endx]<<endl;
    return 0;
}