C4练习题L2-01

L2-001. 紧急救援
时间限制
200 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
作为一个城市的应急救援队伍的负责人，你有一张特殊的全国地图。在地图上显示有多个分散的城市和一些连接城市的快速道路。每个城市的救援队数量和每一条连接两个城市的快速道路长度都标在地图上。当其他城市有紧急求助电话给你的时候，你的任务是带领你的救援队尽快赶往事发地，同时，一路上召集尽可能多的救援队。

输入格式：

输入第一行给出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

题是好题，但是真的很坑，我浪费了好长时间来调bug
迪杰斯特拉算法记录路径数
主要败在num数组上了，当该节点更新最短路径时，应该写num[j] = num[p]，当该节点多出现一条最短路径时，可以num[j]+=num[p];

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
using namespace std;
#define N 550
#define MAXN 0x3f3f3f3f
int mp[N][N];//存图
int path[N];//存路径
int dept[N];//存最短路径长度
int num[N];//存路径个数
int MAX[N];//存最大权和
int a[N];//存每个点的权
int vis[N];//标记数组
int n,m,s,d;
void dijs(int s)
{
     MAX[s] = a[s];
     dept[s] = 0;
    for(int i = 0;i < n;++i)
    {
        if(i != s && mp[i][s] < MAXN){
            dept[i] = mp[i][s];
            MAX[i] = MAX[s] + a[i];
            path[i] = s;
        }
    }
    vis[s] = true;
    for(int i = 0;i < n - 1;++i)
    {
        int MIN = MAXN;
        int p;
        //cout << 0 << endl;
        for(int j = 0;j < n;++j)
        {
            //cout << 0 << endl;
            if(!vis[j] && MIN > dept[j]){
                MIN = dept[j];
                p = j;
            }
        }
        //cout << p << endl;
        vis[p] = true;
       // cout << 0 << endl;
        for(int j = 0;j < n;++j)
        {
            if(!vis[j] && dept[p] + mp[p][j] <= dept[j]){
                if(dept[p] + mp[p][j] == dept[j]){
                        num[j] = num[j] + num[p];
                    if(MAX[p] + a[j] > MAX[j]){
                        MAX[j] = MAX[p] + a[j];
                        path[j] = p;
                    }
                }
                else{
                    MAX[j] = MAX[p] + a[j];
                    path[j] = p;
                    dept[j] = dept[p] + mp[p][j];
                    num[j] = num[p];
                }
            }
        }
    }
}
int main()
{
    scanf("%d %d %d %d",&n,&m,&s,&d);
    memset(vis,false,sizeof(vis));
    memset(mp,MAXN,sizeof(mp));
    memset(MAX,0,sizeof(MAX));
    memset(path,-1,sizeof(path));
    memset(dept,MAXN,sizeof(dept));
    for(int i = 0;i < n;++i)
    {
        scanf("%d",&a[i]);
        num[i] = 1;
    }
    for(int i = 0;i < m;++i)
    {
        int x,y,z;
        scanf("%d %d %d",&x,&y,&z);
        mp[x][y] = min(mp[x][y],z);
        mp[y][x] = min(mp[y][x],z);
    }
    dijs(s);
    cout << num[d] << " " << MAX[d] << endl;
    int x = d;
    int c[N];
    int k = 0;
    while(x != -1)
    {
        c[k++] = x;
        x = path[x];
    }
    for(int i = k - 1;i >= 0;--i)
    {
        if(i == 0){
            cout << c[i] << endl;
        }
        else{
            cout << c[i] << " ";
        }
    }
    return 0;
}