城市间紧急救援 Dijkstra

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

输入格式:

输入第一行给出4个正整数N、M、S、D,其中N(2)是城市的个数,顺便假设城市的编号为0 ~ (;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
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1010;
int n,m,s,d;
int cnt[maxn],vis[maxn],num[maxn],snum[maxn],pre[maxn];
//最短路径条数 是否经过 救援队数量 到达该城市救援队数量 前一个城市
int a[maxn][maxn];

void Dijkstra(){
    vis[s] = 1;cnt[s] = 1;pre[s] = -1;
    for(int i = 0; i < n; i++){
        int min = 0x3f,f = -1;
        for(int j = 0; j < n; j++){
            if(!vis[j] && a[s][j] < min){
                min = a[s][j];
                f = j;
            }
        }
        if(f == -1) break;
        vis[f] = 1;
        for(int j = 0; j < n; j++){
            if(!vis[j] && a[s][j] > a[s][f] + a[f][j]){
                a[s][j] = a[s][f] + a[f][j];
                pre[j] = f;
                cnt[j] = cnt[f];
                snum[j] = num[j] + snum[f];
            }
            else if(!vis[j] && a[s][j] == a[s][f] + a[f][j]){
                cnt[j] += cnt[f];
                if(snum[j] < snum[f] + num[j]){
                    pre[j] = f;
                    snum[j] = snum[f] + num[j];
                }
            }
        }
    }
}
int ans[maxn],tp;
void path(int x){
    ans[tp++] = x;
    while(pre[x] != -1){
        ans[tp++] = pre[x];
        x = pre[x];
    }
}
int main() {
    //freopen("in","r",stdin);
    ios::sync_with_stdio(0);
    cin >> n >> m >> s >> d;
    for(int i = 0; i < n; i++){
        cin >> num[i];
        snum[i] = num[i];
        cnt[i] = 1;
    }
    memset(a,0x3f, sizeof(a));
    for(int i = 0; i < n; i++) a[i][i] = 0;
    while(m--){
        int x,y,z;
        cin >> x >> y >> z;
        a[x][y] = z;a[y][x] = z;
    }
    Dijkstra();
    path(d);
    cout << cnt[d] << " " << snum[d] + num[s] << endl;
    for(int i = tp - 1; i >= 0; i--)
      if(i) cout << ans[i] << " ";
      else cout << ans[i];
    return 0;
}
View Code

 

上一篇:刷题48——最长回文串


下一篇:Redis相关操作