#105270 (C++14) No.160 最短経路のうち辞書順最小

• 問題: No.160 最短経路のうち辞書順最小
• ユーザー: masa
• 提出日時: 2016-07-21 17:11:35
• 言語: C++14 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 16 ms / 5,000 ms
• コード長: 1,790 bytes
• コンパイル時間: 1,217 ms
• コンパイル使用メモリ: 91,416 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-04-23 20:32:05
• 合計ジャッジ時間: 2,352 ms
• ジャッジサーバーID （参考情報）: judge5 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	2 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	5 ms; 5,376 KB
testcase_05	AC	8 ms; 5,376 KB
testcase_06	AC	11 ms; 5,376 KB
testcase_07	AC	3 ms; 5,376 KB
testcase_08	AC	4 ms; 5,376 KB
testcase_09	AC	4 ms; 5,376 KB
testcase_10	AC	4 ms; 5,376 KB
testcase_11	AC	5 ms; 5,376 KB
testcase_12	AC	4 ms; 5,376 KB
testcase_13	AC	4 ms; 5,376 KB
testcase_14	AC	3 ms; 5,376 KB
testcase_15	AC	4 ms; 5,376 KB
testcase_16	AC	3 ms; 5,376 KB
testcase_17	AC	4 ms; 5,376 KB
testcase_18	AC	4 ms; 5,376 KB
testcase_19	AC	3 ms; 5,376 KB
testcase_20	AC	4 ms; 5,376 KB
testcase_21	AC	3 ms; 5,376 KB
testcase_22	AC	4 ms; 5,376 KB
testcase_23	AC	4 ms; 5,376 KB
testcase_24	AC	4 ms; 5,376 KB
testcase_25	AC	4 ms; 5,376 KB
testcase_26	AC	3 ms; 5,376 KB
testcase_27	AC	3 ms; 5,376 KB
testcase_28	AC	16 ms; 5,376 KB
testcase_29	AC	2 ms; 5,376 KB

ソースコード

#include <iostream>
#include <cstdio>
#include <vector>
#include <algorithm>
#include <utility>
#include <queue>
#include <list>

using namespace std;

const int INF = 1e7;

typedef pair<int, int> PII;

vector<vector<int>> dist_matrix;
vector<vector<int>> connect;
vector<int> distances;

void dijkstra(int start) {
	int n = dist_matrix.size();
	distances.assign(n, INF);
	distances[start] = 0;

	priority_queue<PII, vector<PII>, greater<PII>> que;
	que.push(make_pair(0, start));
	while (!que.empty()) {
		auto p = que.top();
		int now_dist = p.first;
		int from = p.second;
		que.pop();
		if (now_dist > distances[from]) {
			continue;
		}

		for (auto to : connect[from]) {
			int d = now_dist + dist_matrix[from][to];
			if (d < distances[to]) {
				distances[to] = d;
				que.push(make_pair(d, to));
			}
		}
	}
}

bool dfs(list<int> &route, int start) {
	int now = route.back();

	for (auto from : connect[now]) {
		if (distances[from] + dist_matrix[from][now] == distances[now]) {
			route.emplace_back(from);
			if (from == start) {
				return true;
			}
			if (dfs(route, start)) {
				return true;
			}
			route.pop_back();
		}
	}
	return false;
}

int main() {
	int n, m, s, g, a, b, c;

	cin >> n >> m >> s >> g;
	dist_matrix.assign(n, vector<int>(n, INF));
	connect.assign(n, vector<int>());

	for (int i = 0; i < m; i++) {
		cin >> a >> b >> c;
		dist_matrix[a][b] = c;
		dist_matrix[b][a] = c;
		connect[a].emplace_back(b);
		connect[b].emplace_back(a);
	}
	for (auto &v : connect) {
		sort(v.begin(), v.end());
	}

	// goalからダイクストラして、startから辞書順に経路復元を試す
	dijkstra(g);

	list<int> route = {s};
	dfs(route, g);

	for (auto x : route) {
		cout << x;
		if (x != g) {
			cout << " ";
		} else {
			cout << endl;
		}
	}
	return 0;
}