L3-011 直捣黄龙 (30 分)-PAT 团体程序设计天梯赛 GPLT

本题是一部战争大片 —— 你需要从己方大本营出发，一路攻城略地杀到敌方大本营。首先时间就是生命，所以你必须选择合适的路径，以最快的速度占领敌方大本营。当这样的路径不唯一时，要求选择可以沿途解放最多城镇的路径。若这样的路径也不唯一，则选择可以有效杀伤最多敌军的路径。

输入格式：

输入第一行给出 2 个正整数 N（2 ≤ N ≤ 200，城镇总数）和 K（城镇间道路条数），以及己方大本营和敌方大本营的代号。随后 N-1 行，每行给出除了己方大本营外的一个城镇的代号和驻守的敌军数量，其间以空格分隔。再后面有 K 行，每行按格式城镇1 城镇2 距离给出两个城镇之间道路的长度。这里设每个城镇（包括双方大本营）的代号是由 3 个大写英文字母组成的字符串。

输出格式：

按照题目要求找到最合适的进攻路径（题目保证速度最快、解放最多、杀伤最强的路径是唯一的），并在第一行按照格式己方大本营->城镇1->...->敌方大本营输出。第二行顺序输出最快进攻路径的条数、最短进攻距离、歼敌总数，其间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

10 12 PAT DBY
DBY 100
PTA 20
PDS 90
PMS 40
TAP 50
ATP 200
LNN 80
LAO 30
LON 70
PAT PTA 10
PAT PMS 10
PAT ATP 20
PAT LNN 10
LNN LAO 10
LAO LON 10
LON DBY 10
PMS TAP 10
TAP DBY 10
DBY PDS 10
PDS PTA 10
DBY ATP 10

输出样例：

PAT->PTA->PDS->DBY
3 30 210

分析：本题相对比较简单，只需要根据题目要求调整最短路径的判定条件就好，为了方便操作，这里大本营代号（字符串）与数值之间要做一个映射～ntoi保存名字到编号的转换，iton保存编号到名字的转换，num中储存各个大本营敌军数量，Num中存储歼敌总数，path中存储每个大本营由哪个地方过来的，Dis存储出发点到各点的最短距离，sum中存储路径数，liberation存储了解放的大本营个数～

#include <bits/stdc++.h>
using namespace std;
struct node {
	int id, dis;
	bool friend operator < (const node &a, const node &b) {
		return a.dis > b.dis;
	}
};
int n, k, s, t, cnt, d, num[205], Num[205], path[205], Dis[205], vis[205], sum[205], liberation[205];
map<string, int> ntoi;
map<int, string> iton;
vector<pair<int, int>> E[205];
vector<int> ans;
string S, T, u, v;
void Dijskra() {
	fill(Dis, Dis + 205, 1e9);
	Dis[s] = 0;
	sum[s] = 1;
	priority_queue<node> Q;
	Q.push(node{s, 0});
	while (!Q.empty()) {
		int Now = Q.top().id, D = Q.top().dis;
		Q.pop();
		if (vis[Now]) continue;
		vis[Now] = 1;
		for (auto it : E[Now]) {
			int V = it.first, DIS = it.second;
			if (Dis[V] > D + DIS) {
				Num[V] = Num[Now] + num[V];
				sum[V] = sum[Now];
				liberation[V] = liberation[Now] + 1;
				Dis[V] = D + DIS;
				Q.push(node{V, Dis[V]});
				path[V] = Now;
			} else if (Dis[V] == D + DIS) {
				sum[V] += sum[Now];
				if (liberation[V] < liberation[Now] + 1) {
					liberation[V] = liberation[Now] + 1;
					Num[V] = Num[Now] + num[V];
					Q.push(node{V, Dis[V]});
					path[V] = Now;
				} else if (liberation[V] == liberation[Now] + 1) {
					if (Num[V] < Num[Now] + num[V]) {
						Num[V] = Num[Now] + num[V];
						Q.push(node{V, Dis[V]});
						path[V] = Now;
					}
				}
				
			}
		}
	}
}
int main() {
	cin >> n >> k >> S >> T;
	for (int i = 0; i < n - 1; i++) {
		cin >> u >> d;
		ntoi[u] = ++cnt;
		iton[cnt] = u;
		num[cnt] = d;
	}
	ntoi[S] = 0, iton[0] = S, t = ntoi[T];
	for (int i = 0; i < k; i++) {
		cin >> u >> v >> d;
		E[ntoi[u]].push_back({ntoi[v], d});
		E[ntoi[v]].push_back({ntoi[u], d});
	}
	Dijskra();
	d = t;
	while(d) {
		ans.push_back(d);
		d = path[d];
	}
	cout << iton[0];
	for (int i = ans.size() - 1; i >= 0; i--) cout << "->" << iton[ans[i]];
	cout << '\n' << sum[t] << ' ' << Dis[t] << ' ' << Num[t] << '\n';
	return 0;
}

#include <bits/stdc++.h>

using namespace std;

struct node {

int id, dis;

bool friend operator < (const node &a, const node &b) {

return a.dis > b.dis;

}

};

int n, k, s, t, cnt, d, num[205], Num[205], path[205], Dis[205], vis[205], sum[205], liberation[205];

map<string, int> ntoi;

map<int, string> iton;

vector<pair<int, int>> E[205];

vector<int> ans;

string S, T, u, v;

void Dijskra() {

fill(Dis, Dis + 205, 1e9);

Dis[s] = 0;

sum[s] = 1;

priority_queue<node> Q;

Q.push(node{s, 0});

while (!Q.empty()) {

int Now = Q.top().id, D = Q.top().dis;

Q.pop();

if (vis[Now]) continue;

vis[Now] = 1;

for (auto it : E[Now]) {

int V = it.first, DIS = it.second;

if (Dis[V] > D + DIS) {

Num[V] = Num[Now] + num[V];

sum[V] = sum[Now];

liberation[V] = liberation[Now] + 1;

Dis[V] = D + DIS;

Q.push(node{V, Dis[V]});

path[V] = Now;

} else if (Dis[V] == D + DIS) {

sum[V] += sum[Now];

if (liberation[V] < liberation[Now] + 1) {

liberation[V] = liberation[Now] + 1;

Num[V] = Num[Now] + num[V];

Q.push(node{V, Dis[V]});

path[V] = Now;

} else if (liberation[V] == liberation[Now] + 1) {

if (Num[V] < Num[Now] + num[V]) {

Num[V] = Num[Now] + num[V];

Q.push(node{V, Dis[V]});

path[V] = Now;

}

int main() {

cin >> n >> k >> S >> T;

for (int i = 0; i < n - 1; i++) {

cin >> u >> d;

ntoi[u] = ++cnt;

iton[cnt] = u;

num[cnt] = d;

}

ntoi[S] = 0, iton[0] = S, t = ntoi[T];

for (int i = 0; i < k; i++) {

cin >> u >> v >> d;

E[ntoi[u]].push_back({ntoi[v], d});

E[ntoi[v]].push_back({ntoi[u], d});

}

Dijskra();

d = t;

while(d) {

ans.push_back(d);

d = path[d];

}

cout << iton[0];

for (int i = ans.size() - 1; i >= 0; i--) cout << "->" << iton[ans[i]];

cout << '\n' << sum[t] << ' ' << Dis[t] << ' ' << Num[t] << '\n';

return 0;

}

❤ 点击这里 -> 订阅《PAT | 蓝桥 | LeetCode学习路径 & 刷题经验》by 柳婼

❤ 点击这里 -> 订阅《从放弃C语言到使用C++刷算法的简明教程》by 柳婼

❤ 点击这里 -> 订阅PAT甲级乙级、蓝桥杯、GPLT天梯赛、LeetCode题解离线版