【题解】PAT 1003 Emergency

题目描述

原题链接：1003 Emergency

解题思路

首先读题，这是一个图论的问题，需要从一个城市走最短路径到另一个城市（最短路径可能不止一条），每个城市都会有一些人手，同时要求求出走这些最短路径最多召集多少人手。
要求输出最短路径的条数和最多召集多少人手。

---

首先解决要求出最短路径的条数，这个只需要在Dijkstra算法上做一些简单的改动即可，回顾一下最经典的Dijkstra算法：

起点城市为start
建立一个数组dis（dis[i]表示从start城市到i城市的距离），将其中所有元素初始化为无穷大，dis[start]=0
建立一个数组vis（vis[i]表示城市i是否处于队列中），所有元素初始化为false

1. 遍历dis数组，找到一个下标i，满足vis[i]=false且dis[i]最小
2. 遍历以节点i为起点的所有边(i, to, weight)，dis[to]=min(dis[i]+weight, dis[to])
3. 重复上述步骤，知道所有节点遍历（实际上只需要遍历n-1次）

---

要求出最短路径的条数，只需稍微修改一下即可：

新建一个数组num，num[i]用来存储从起点k到节点i的最短路径条数

1. 遍历dis数组，找到一个下标i，满足vis[i]=false且dis[i]最小
2. 遍历以节点i为起点的所有边(i, to, weight)，dis[to]=min(dis[i]+weight, dis[to])
   如果dis[to]>dis[i]+weight，则num[i]=num[to], dis[to]=dis[i]+weight
   如果dis[to]==dis[i]+weight，则num[i]+=num[to]
3. 重复上述步骤，知道所有节点遍历（实际上只需要遍历n-1次）

---

同时还可以求出最多召集的人手数：

数组city，其中city[i]表示城市i中可召集的人手数
数组hand，其中hand[i]表示沿着最短路径到达城市i最多可召集的人手数

1. 遍历dis数组，找到一个下标i，满足vis[i]=false且dis[i]最小
2. 遍历以节点i为起点的所有边(i, to, weight)，dis[to]=min(dis[i]+weight, dis[to])
   如果dis[to]>dis[i]+weight，则num[i]=num[to], dis[to]=dis[i]+weight且hand[to]=hand[i]+city[to]
   如果dis[to]==dis[i]+weight，则num[i]+=num[to]且hand[to]=max(hand[to], hand[i]+city[to])
3. 重复上述步骤，知道所有节点遍历（实际上只需要遍历n-1次）

综上，对于目标城市target，只需要输出num[target], hand[target]

AC代码

#include <bits/stdc++.h>
using namespace std;
const int num_node = 501;

struct Edge {
    int to, weight;
    Edge(int to, int weight):to(to), weight(weight) {}
};
vector<Edge> edge[num_node];

void init(int n) {
    for (int i=0;i<n;i++) edge[i].clear();
}

void addEdge(int from, int to, int weight) {
    edge[from].push_back(Edge(to, weight));
}

int main () {
    int city[num_node], n, m, start, target;
    scanf("%d %d %d %d", &n, &m, &start, &target); init(n);
    for(int i=0;i<n;i++) scanf("%d", &city[i]);
    for(int i=0;i<m;i++) {
        int from, to, weight;
        scanf("%d %d %d", &from, &to, &weight);
        addEdge(from, to, weight);
        addEdge(to, from, weight);
    }

    int dis[num_node], num[num_node], hand[num_node]; bool vis[num_node];
    memset(dis, 127, sizeof(dis)); dis[start]=0;  // 将dis初始化为最大值
    memset(num, 0, sizeof(num)); num[start] = 1;
    memset(hand, 0, sizeof(hand)); hand[start] = city[start];
    memset(vis, 0, sizeof(vis));
    
    for (int count = 1; count < n; count ++) {
        int index=0, flag = INT_MAX;
        for (int i = 0; i < n; i++) if (!vis[i] && dis[i] < flag) {
                index = i; flag = dis[index];
        }
        vis[index] = true;

        for (int i = 0; i < edge[index].size(); i++) {
            if (dis[edge[index][i].to] > dis[index] + edge[index][i].weight) {
                dis[edge[index][i].to] = dis[index] + edge[index][i].weight;
                num[edge[index][i].to] = num[index];
                hand[edge[index][i].to] = hand[index] + city[edge[index][i].to];
            } else if (dis[edge[index][i].to] == dis[index] + edge[index][i].weight) {
                num[edge[index][i].to] += num[index];
                hand[edge[index][i].to] = max(hand[edge[index][i].to], hand[index] + city[edge[index][i].to]);
            }
        }
    }

    printf("%d %d\n", num[target], hand[target]);
//    printf("%d %d %d\n", start, target, dis[target]);
    return 0;
}