L2-001 紧急救援

该题为双元素最短路问题

所要保存最小首元素、最大第二元素、目标最短路路径，最短路条数

#include <bits/stdc++.h>
#define endl '\n'
#define ll long long
#define INF 0x3f3f3f3f
using namespace std;
const int N = 510;
typedef pair<int, int> PII;
struct Node // 用于邻接表的点结构体
{
    int v, w; // 代表当前结点可以通向v结点，边权为w
    int k;    // v结点的救援队数量
};
vector<Node> g[N]; // 邻接表
int dist[N];       // 每个点到初始点的最小距离
bool vis[N];       // 每个点是否被松弛过
int num[N];        // 存储每个点到初始点的最短路条数
int sum[N];        // 存储每个点的救援队数量
int K[N];          // 存储每个点到初始点积累的最大救援队数量
// 由于所要求的是救援队最大数量，所以不需要初始化，使用默认零即可
int path[N]; // 记录每个点的前驱结点用于存储目标路径
int n, m, s, t;
void init() // 初始化
{
    for (int i = 0; i < n; i++)
    {
        dist[i] = INF;
        vis[i] = false;
    }
}
void dij()
{
    dist[s] = 0;   // 初始点最小距离为零
    K[s] = sum[s]; // 初始点的最大救援队数量即为该点的救援队数量
    num[s] = 1;    // 初始点的最短路条数为1
    path[s] = s;   // 标记初始点的前去结点为本身
    priority_queue<PII, vector<PII>, greater<PII>> q;
    q.push({dist[s], s}); // 按照最小距离进行排序
    while (q.size())
    {
        int u = q.top().second;
        q.pop();
        if (vis[u])
            continue;
        vis[u] = true;
        for (auto [v, w, k] : g[u])
        {
            if (dist[v] > dist[u] + w) // 距离需要更新
            {
                dist[v] = dist[u] + w; // 更新距离
                K[v] = K[u] + k;       // 更新最大救援队数量
                num[v] = num[u];       // 更新最短路条数采用直接赋值，因为找到了更小的最短路
                path[v] = u;           // 记录该点的前驱结点
                q.push({dist[v], v});
            }
            else if (dist[v] == dist[u] + w) // 距离相等
            {
                if (K[v] < K[u] + k) // 最大救援队数量需要更新
                {
                    K[v] = K[u] + k; // 取更大的最大救援队熟练
                    path[v] = u;     // 记录前驱节点
                    q.push({dist[v], v});
                }
                num[v] += num[u]; // 由于两点的最小距离相等
                // 所以将最短路条数直接相加
            }
        }
    }
}
void Print(int t) // 由于path里存储的是前驱结点
// 所以要递归输出
{
    if (t == path[t]) // 边界条件为初始点
    {
        cout << t;
        return;
    }
    Print(path[t]); // 输出完初始结点之后开始递归输出前驱结点
    cout << " " << t;
}
int main()
{
    cin >> n >> m >> s >> t;
    init();
    for (int i = 0; i < n; i++)
    {
        cin >> sum[i];//存储每个结点的救援队数量
    }
    for (int i = 0; i < m; i++)//建立邻接表
    {
        int u, v, w;
        cin >> u >> v >> w;
        g[u].push_back({v, w, sum[v]});
        g[v].push_back({u, w, sum[u]});
    }
    dij();
    cout << num[t] << " " << K[t] << endl;
    Print(t);
    return 0;
}