CCCC-L2-001 紧急救援

题目

作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式:

输入第一行给出4个正整数N、M、S、D，其中N（2≤N≤500）是城市的个数，顺便假设城市的编号为0 ~ (N−1)；M是快速道路的条数；S是出发地的城市编号；D是目的地的城市编号。

第二行给出N个正整数，其中第i个数是第i个城市的救援队的数目，数字间以空格分隔。随后的M行中，每行给出一条快速道路的信息，分别是：城市1、城市2、快速道路的长度，中间用空格分开，数字均为整数且不超过500。输入保证救援可行且最优解唯一。

输出格式:

第一行输出最短路径的条数和能够召集的最多的救援队数量。第二行输出从S到D的路径中经过的城市编号。数字间以空格分隔，输出结尾不能有多余空格。

输入样例:

4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2

输出样例:

2 60
0 1 3

代码

#include <bits/stdc++.h>

using namespace std;

struct city{
    int helpNum;
    int sumHelpNum;
    int pre;//前导
    int dist;
    int minNum;
    bool visited;
};
int n,m,s,d,Map[550][550];
city arr[550];


void input()
{
    cin>>n>>m>>s>>d;
    for(int i=0;i<n;i++)
    {
        cin>>arr[i].helpNum;
        arr[i].sumHelpNum=arr[i].helpNum;//初始化-------
    }
    int a,b,c;
    for(int i=0;i<m;i++)
    {
        cin>>a>>b>>c;
        Map[a][b]=c; Map[b][a]=c;//----------------------------
    }
}

void dijkstra()
{
    int now=s;
    arr[now].dist=0;
    arr[now].visited=true;
    arr[now].minNum=1;

    //找初始now
    int min_dist=1000000;
    /*for(int i=0;i<n;i++)
    {
        if(Map[s][i]<min_dist&&i!=s){min_dist=Map[s][i]; now=i; }
    }*/
    //循环加入除s外其他的点
    for(int i=0;i<n;i++)
    {
        //更新
        for(int j=0;j<n;j++){
            if(arr[j].dist>arr[now].dist+Map[now][j])
            {
                arr[j].dist=arr[now].dist+Map[now][j];
                arr[j].pre=now;
                arr[j].sumHelpNum=arr[now].sumHelpNum+arr[j].helpNum;
                arr[j].minNum=arr[now].minNum;//--------------------------------
            }
            else if(now!=j&&Map[now][j]!=1000000&&arr[j].dist==arr[now].dist+Map[now][j])//最短路径相同，救援队数目不同
            {
                arr[j].minNum+=arr[now].minNum;
                if(arr[j].sumHelpNum<arr[now].sumHelpNum+arr[j].helpNum)
                {
                    arr[j].dist=arr[now].dist+Map[now][j];
                    arr[j].pre=now;
                    arr[j].sumHelpNum=arr[now].sumHelpNum+arr[j].helpNum;
                }
            }
        }
        //选出下一个now
        min_dist=1000000;
        for(int j=0;j<n;j++)
        {
            if(arr[j].visited==false&&arr[j].dist<min_dist)
            {
                now=j;
                min_dist=arr[j].dist;
            }
        }
        arr[now].visited=true;
    }
}

void pre(int d)
{
    if(d==-1) return;//结束条件
    pre(arr[d].pre);
    cout<<d<<" ";
}

void output()
{
    cout<<arr[d].minNum<<" "<<arr[d].sumHelpNum<<endl;

    pre(arr[d].pre);
    cout<<d;
}

int main()
{
    //初始化
    for(int i=0;i<550;i++)
    {
        for(int j=0;j<550;j++)
            Map[i][j]=1000000;
        Map[i][i]=0;//------------------------------------------

        arr[i].visited=false;
        arr[i].pre=-1;
        arr[i].dist=1000000;
    }

    input();
    dijkstra();
    output();
    return 0;
}

/*4 5 0 3
20 30 40 10
0 1 1
1 3 2
0 3 3
0 2 2
2 3 2*/
/*
5 5 0 3
30 100 20 50 120
0 1 1
1 2 2
2 3 3
4 3 5
0 4 1
*/

总结

• 就是一个Dijkstra算法的简单应用+变形
• 需要注意的是最短路径数的求法
  • 最短路径数的更新和最短路径路线的更新是不同步的，需要分开写
  • 第一种情况：路径长度小于原有值进行更新——最短路径数=前一节点最短路径数
  • 第二种情况：路径长度等于原有值进行更新（救援队数量）——最短路径数+=前一节点最短路径数
• 使用结构体组织比较清晰
• 样例来源：样例