PAT A1003 Emergency

一、题目详情

PAT 1003题目
相当于给定一个无向图，边权和点权，求出最短路径的条数和最短路径中点权之和的最大值

输入解释

• 第一行有四个整数：N、M、C1、C2，分别为图中的节点数，边数，起点编号和终点编号
• N个整数，依次为节点的点权
• 接下来M行为节点的端点编号 边权

输出解释

• 两个数字，分别为最短路径的条数和最短路径中点权之和的最大值

思路

• 固定起点和终点，典型的Dijkstra算法
• 题目中还需要求出点权之和最大值，因此可以用一个数组w，w[u]表示起点s到顶点u可以得到的最大点权之和
• 每次在进行“松弛”操作时，如果松弛成功，就覆盖d[v]和w[v]，如果松弛/不松弛结果相同的话，就看：如果进行松弛操作，能否使w[v]更大，如果能，就覆盖w[v]。相当于路径长度是第一标尺，点权之和是第二标尺。
• 题目还要输出最短路径的条数，需要再添加一个数组num，num[u]表示从s到u的最短路径的条数，每次如果松弛成功，相当于u到v这条边被选中，那么num[v]=num[u]，如果松弛/不松驰结果相同，那么相当于从s到v的最短路径数等于从s到v（不经过u）的最短路径数加上从s到u到v的最短路径数，即num[v]+=num[u]。

二、实现代码

#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int maxn = 510;
const int inf = 10000000;
int n, m, c1, c2;
int G[maxn][maxn];
int weight[maxn];
int d[maxn], w[maxn], num[maxn];
bool visited[maxn] = {false};
void dijkstra(int s)
{
    int min;
    fill(num, num + maxn, 0);
    fill(w, w + maxn, 0);
    fill(d, d + maxn, inf);
    d[s] = 0;
    w[s] = weight[s];
    num[s] = 1;
    for (int i = 0; i < n; i++)
    {
        int u = -1;
        min = inf;
        for (int j = 0; j < n; j++)
        {
            if (d[j] < min && !visited[j])
            {
                u = j;
                min = d[j];
            }
        }
        if (u == -1)
            return;
        visited[u] = true;
        for (int k = 0; k < n; k++)
        {
            if (!visited[k] && G[u][k] != inf)
            {
                if (d[u] + G[u][k] < d[k])
                {
                    d[k] = d[u] + G[u][k];
                    w[k] = w[u] + weight[k];
                    num[k] = num[u];
                }
                else if (d[u] + G[u][k] == d[k])
                {
                    if (w[u] + weight[k] > w[k])
                        w[k] = w[u] + weight[k];

                    num[k] += num[u];
                }
            }
        }
    }
}
int main()
{
    int u, v;
    fill(G[0], G[0] + maxn * maxn, inf);
    scanf("%d%d%d%d", &n, &m, &c1, &c2);
    for (int i = 0; i < n; i++)
    {
        scanf("%d", &weight[i]);
    }
    for (int i = 0; i < m; i++)
    {
        scanf("%d%d", &u, &v);
        scanf("%d", &G[u][v]);
        G[v][u] = G[u][v];
    }
    dijkstra(c1);
    cout << num[c2] << " " << w[c2] << endl;
    return 0;
}