L2-001. 紧急救援

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

最短路问题，但是有两个最优先度判断，第一优先度是路程，第二优先度是人数；

Dijkstra算法代码如下：

//
//  main.cpp
//  L2-1. 紧急救援
//
//  Created by 徐智豪 on 2017/4/8.
//  Copyright © 2017年 徐智豪. All rights reserved.
//

#include <iostream>
using namespace std;
int n,m,s,d;
int people[505],map[505][505],way[505],vis[505],sum[505],pre[505],dis[505];
void print( int temp)//get the pre city
{
    if(pre[temp]!=-1)
    {
        print(pre[temp]);
        cout<<pre[temp]<<" ";
    }
}
int main(int argc, const char * argv[]) {
    cin>>n>>m>>s>>d;
    for(int i=0;i<n;i++)
        cin>>people[i];
    int x,y,z;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
        {
            if(i==j)map[i][j]=0;
            else map[i][j]=1000000;
        }
    for(int i=0;i<m;i++)
    {
        cin>>x>>y>>z;
        map[x][y]=map[y][x]=z;
    }
    for(int i=0;i<n;i++)
    {
        vis[i]=0;
        sum[i]=0;
        pre[i]=-1;
        way[i]=0;
        dis[i]=1000000;
    }
    sum[s]=people[s];
    way[s]=1;
    vis[s]=1;
    dis[s]=0;
    for(int i=0;i<n;i++)
    {
        int thecity=s,mindis=1000000;
        for(int j=0;j<n;j++)
        {
            if(vis[j]==0&&mindis>dis[j])
            {
                thecity=j;
                mindis=dis[j];
            }
        }
        vis[thecity]=1;
        for(int j=0;j<n;j++)
        {
            if(vis[j]==0)
            {
                if(dis[j]>dis[thecity]+map[thecity][j])
                {
                     dis[j]=dis[thecity]+map[thecity][j];
                sum[j]=sum[thecity]+people[j];
                pre[j]=thecity;
                way[j]=way[thecity];
                }
                else if(dis[j]==(dis[thecity]+map[thecity][j]))
                {
                    if(sum[j]<sum[thecity]+people[j])
                    {
                        sum[j]=sum[thecity]+people[j];
                        pre[j]=thecity;
                    }
                    way[j]+=way[thecity];
                }
            }
        }
    }
    cout<<way[d]<<" "<<sum[d]<<endl;
    print( d);
    cout<<d<<endl;
    return 0;
}