【PAT 甲级】1003 Emergency (25分)（dfs + STL）

1003 Emergency (25分)

题目

给一个顶点数不超过 500 的图，每个顶点有权值，边也有权值。给定起点和终点，求边权值最短的路径集合（也就是说最短的路径不止一条，而且每条最短路径对应一个顶点权值和），输出这个集合的路径个数，以及集合中顶点和的最大值。

分析

题目在求最短路径的基础上，还增加了每条路径对应的顶点权值和。

所以可以用 dfs 求出所有最短路径，并将对应的权值和放入到 multiset 中，数据结构如下：
map<int, multiset<int> > mp
mp[k] 表示长度为 k 的最短路径对应的顶点权值集合。

dfs 过程中可以设置一个当前最短路径值 min_len，如果到当前递归为止的路径值大于 min_len，就 return，算是剪枝。

实现当中要求 multiset 中的最大值，也就是最后一个元素，可以用反向迭代器 rbegin()

#include <bits/stdc++.h>
using namespace std;
#define db(x) cout<<x<<endl

typedef long long ll;
const int INF = 0x3f3f3f3f;
const int N = 5e2 + 10;
const ll mod = 2147483648;

int n, m, c1, c2;
int num[N], a[N][N], vis[N];
int min_len = INF;
map<int, multiset<int> > mp; // m[k]，道路长度为 k 的所有方案人数集合
map<int, multiset<int> >::iterator it;

void dfs(int rt, int len, int men) {
	if (rt == c2) {
		if (len <= min_len) {
			min_len = len;
			mp[len].insert(men);
		}
		return;
	}
	if (len > min_len) return;
	for (int i = 0; i < n; i++) {
		if (!vis[i] && a[rt][i] != INF) {
			vis[i] = 1;
			dfs(i, len + a[rt][i], men + num[i]);
			vis[i] = 0;
		}
	}
}

int main() {
	scanf("%d%d%d%d", &n, &m, &c1, &c2);
	for (int i = 0; i < n; i++) scanf("%d", &num[i]);
	memset(a, INF, sizeof(a));
	for (int i = 0, x, y, z; i < m; i++) {
		scanf("%d%d%d", &x, &y, &z);
		a[x][y] = a[y][x] = z;
	}
	vis[c1] = 1;
	dfs(c1, 0, num[c1]);
	it = mp.begin();
	printf("%d %d\n", it->second.size(), *(it->second.rbegin()));
	return 0;
}