天梯 L2-优快云博客

本文链接：https://blog.youkuaiyun.com/lareinaba/article/details/124284403

L2-001 紧急救援 Dijkstra

在这里插入图片描述

#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,c1,c2;
int dis[510],weight[510],e[510][510],num[510],w[510],pre[510];//最短路径长度，救援队数量，两城市之间的路的长度，用来更新最短路径数量的数组，用来更新最多救援队数量的数组，记录前驱的数组
bool visit[510];
const int inf=99999999;

void printPath(int v)//打印从起点出发到终点的路径
{
    if(v==c1){
        printf("%d",v);
        return;
    }
    printPath(pre[v]);
    printf(" %d",v);
}

int main()
{
    scanf("%d%d%d%d",&n,&m,&c1,&c2);
    for(int i=0;i<n;i++)
        scanf("%d",&weight[i]);//每个城市救援队的数量
    fill(e[0],e[0]+510*510,inf);
    fill(dis,dis+510,inf);
    int a,b,c;
    for(int i=0;i<m;i++){//城市1，城市2，路径长度
        scanf("%d%d%d",&a,&b,&c);
        e[a][b]=c;
        e[b][a]=c;
    }
    dis[c1]=0;
    w[c1]=weight[c1];//用w[]来更新最多救援队数量
    num[c1]=1;

    for(int i=0;i<n;i++){

        int u=-1,minn=inf;
        for(int j=0;j<n;j++){//求出最近的点
            if(visit[j]==false&&dis[j]<minn){
                u=j;
                minn=dis[j];
            }
        }

        if(u==-1)break;
        visit[u]=true;//将点u划入s集合（源点集）
        for(int v=0;v<n;v++)
        {
            if(visit[v]==false&&e[u][v]!=inf){//没有被访问过且存在路
                if(dis[u]+e[u][v]<dis[v]){//松弛操作
                    dis[v]=dis[u]+e[u][v];
                    num[v]=num[u];//最短路径的数量
                    w[v]=w[u]+weight[v];//救援队的数量
                    pre[v]=u;//前驱
                }else if(dis[u]+e[u][v]==dis[v])//如果路径相等，就比较救援队数量
                {
                    num[v]=num[v]+num[u]; //最短路径（长度相等和可能不止一种）的条数
                    if(w[u]+weight[v]>w[v]){
                        w[v]=w[u]+weight[v];
                        pre[v]=u;
                    }
                }
            }
        }

    }
    printf("%d %d\n",num[c2],w[c2]);
    printPath(c2);
    return 0;
}