PAT 团体天梯赛 L2-001 紧急救援 (迪杰斯特拉)

题意： 中文的就不说了

方法：方法一看就知道 单源最短路径并且维护另外的两个救援人数和方法数就好了 因为习惯了用spfa写 怎么调试都是部分正确 7分的那组测试数据就是过不了 硬是想不到为什么

后来用迪杰斯特拉跑了一发就过了  后来仔细想了想  迪杰斯特拉对于那些已经选择的点是不会去更新的 spfa虽然能跑出正确的最短路径 但是维护那两个量会出现一点问题 毕竟只

有保证选择的这个点是最短路径才去更新那两个量才可以 而spfa那种一松弛就更新那两个量就会出现问题。这也个人看法 仅供参考 如果有用spfa过的 一定要分享一下你的思路

#include <iostream>
#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <string>
#include <map>
#include <math.h>
#include <queue>
#include <climits>
#include <stack>
using namespace std;
int n, m, s, d;
const int maxn = 505;
int ss[maxn];
int mm[maxn][maxn];
int sum[maxn];
int dis[maxn];
int vis[maxn];
int pre[maxn];
int cnt[maxn];
void path(int x)
{
    if(pre[x] != -1){
        path(pre[x]);
        cout<<pre[x]<<' ';
    }
}
void init()
{
    for(int i = 0; i < n; i++)
    {
        vis[i] = 0;
        sum[i] = 0;
        dis[i] = INT32_MAX;
        pre[i] = -1;
        cnt[i] = 0;
    }
}
void djstla(int op, int ed)
{
    init();
    vis[op] = 1;
    sum[op] += ss[op];
    dis[op] = 0;
    cnt[op] = 1;
    for(int i = 0; i < n; i++){
        int min_key = INT_MAX, min_pos = op;
        for(int j = 0; j < n; j++){
            if(!vis[j] && min_key > dis[j]){
                min_key = dis[j];
                min_pos = j;
            }
        }
        vis[min_pos] = 1;
        if(min_pos == ed) return;
        for(int j = 0; j < n; j++){
            if(mm[min_pos][j] != -1 && !vis[j]){
                if(dis[min_pos] + mm[min_pos][j] < dis[j]){
                    dis[j] = dis[min_pos] + mm[min_pos][j];
                    sum[j] = sum[min_pos] + ss[j];
                    cnt[j] = cnt[min_pos];
                    pre[j] = min_pos;
                }
                else if(dis[min_pos] + mm[min_pos][j] == dis[j]){
                    cnt[j] += cnt[min_pos];
                    if(sum[j] < sum[min_pos] + ss[j]){
                        sum[j] = sum[min_pos] + ss[j];
                        pre[j] = min_pos;
                    }
                }
            }
        }
    }
}


int main()
{
    
    cin>>n>>m>>s>>d;
    for(int i = 0; i < n; i++)
        cin>>ss[i];
    memset(mm, -1, sizeof mm);
    for(int i = 0; i < m; i++)
    {
        int a, b, c;
        cin>>a>>b>>c;
        if(mm[a][b] == -1)
        {
            mm[a][b] = c;
            mm[b][a] = c;
        }
        else{
            mm[a][b] = min(mm[a][b], c);
            mm[b][a] = min(mm[b][a], c);
        }
    }
    djstla(s, d);
    cout<<cnt[d]<<' '<<sum[d]<<endl;
    path(d);
    cout<<d<<endl;
    return 0;
}