PTA 一 直搗黃龍

本題是一部戰爭大片 —— 你需要從己方大本營出發,一路攻城略地殺到敵方大本營。首先時間就是生命,所以你必須選擇合適的路徑,以最快的速度佔領敵方大本營。當這樣的路徑不唯一時,要求選擇可以沿途解放最多城鎮的路徑。若這樣的路徑也不唯一,則選擇可以有效殺傷最多敵軍的路徑。

輸入格式:

輸入第一行給出2個正整數N(2 \le N \le 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

算法思路:利用字典使字符串和下標產生一一對應關係。單源最短路徑拓展。

#include<stdio.h>
#include<map>
#include<string>
#include<stack>
using namespace std;
map<string, int> match;
const int MAX = 100000;
int army[200];
int MAP[200][200];
bool vis[200];
int dis[200];
int arms[200];
int citycount[200];
int pre[200];
int waycount[200];
string NAMES[200];

void Dijkstra(int d, int n) {
	int i;
	for (i = 0; i < n; i++) dis[i] = MAX, vis[i] = false;
	int temp;
	int u = 0;
	dis[u] = 0; citycount[u] = 0; waycount[u] = 1;
	while (u != d) {
		vis[u] = true;
		for (i = 0; i < n; i++) {
			if (vis[i] || MAP[u][i] == MAX) continue;
			temp = MAP[u][i] + dis[u];
			if (temp < dis[i]) {
				dis[i] = temp;
				pre[i] = u;
				arms[i] = arms[u] + army[i];
				citycount[i] = citycount[u] + 1;
				waycount[i] = waycount[u];
			}
			else if (temp == dis[i]) {
				waycount[i] += waycount[u];
				temp = citycount[u] + 1;
				if (temp > citycount[i]) {
					citycount[i] = temp;
					pre[i] = u;
					arms[i] = arms[u] + army[i];
				}
				else if (temp == citycount[i]) {
					temp = arms[u] + army[i];
					if (temp > arms[i]) {
						arms[i] = temp;
						pre[i] = u;
					}
				}
			}
		}
		temp = MAX;
		for (i = 0; i < n; i++) {
			if (vis[i]) continue;
			if (temp > dis[i]) {
				temp = dis[i];
				u = i;
			}
		}
	}
}


int main() {
	int n, k;
	scanf("%d%d", &n, &k);
	string sName, dName;
	sName.resize(4);
	dName.resize(4);
	getchar();
	scanf("%s %s", &sName[0], &dName[0]);
	match.insert(pair<string, int>(sName, 0));
	int i;
	string name; int value;
	name.resize(4);
	for (i = 1; i < n; i++) {
		scanf("%s %d", &name[0], &value);
		match[name] = i;
		NAMES[i] = name;
		army[i] = value;
	}
	string c1, c2; int distance, ic1, ic2;
	c1.resize(4);
	c2.resize(4);
	for (i = 0; i < n; i++) for (int j = 0; j < n; j++) MAP[i][j] = MAX;
	for (i = 0; i < k; i++) {
		scanf("%s %s %d", &c1[0], &c2[0], &distance);
		ic1 = match[c1], ic2 = match[c2];
		MAP[ic1][ic2] = distance;
		MAP[ic2][ic1] = distance;
	}
	int dest = match[dName];
	Dijkstra(dest, n);
	i = dest;
	stack<int> sta;
	while (i != 0) {
		sta.push(i);
		i = pre[i];
	}
	printf("%s", sName.c_str());
	while (!sta.empty()) {
		printf("->%s", NAMES[sta.top()].c_str());
		sta.pop();
	}
	printf("\n%d %d %d\n", waycount[dest], dis[dest], arms[dest]);
	return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章