結果

問題 No.160 最短経路のうち辞書順最小
ユーザー やまぞう
提出日時 2015-04-09 01:11:51
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 27 ms / 5,000 ms
コード長 1,408 bytes
コンパイル時間 907 ms
コンパイル使用メモリ 81,204 KB
実行使用メモリ 5,888 KB
最終ジャッジ日時 2024-07-04 13:20:50
合計ジャッジ時間 2,090 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <algorithm>
#include <climits>
#include <iostream>
#include <map>
#include <vector>

int N;
int M;
int S;
int G;
std::map<std::pair<int, int>, int> cost;
const int INF = 2000000;
std::vector<int> dist;
std::vector<bool> used;
std::vector<int> next;

int getCost(int u, int v)
{
	if (cost.find(std::pair<int, int>(u, v)) == cost.end()) {
		return INF;
	}
	return cost[std::pair<int, int>(u, v)];
}

void set_nmsg(int n, int m, int s, int g)
{
	N = n;
	M = m;
	S = s;
	G = g;
}

void set_abc(int a, int b, int c)
{
	cost[std::pair<int, int>(a, b)] = c;
	cost[std::pair<int, int>(b, a)] = c;
}

void resolve()
{
	dist.resize(N);
	used.resize(N);
	next.resize(N);

	std::fill(dist.begin(), dist.end(), INF);
	std::fill(used.begin(), used.end(), false);
	std::fill(next.begin(), next.end(), -1);
	dist[G] = 0;

	for (;;) {
		int v = -1;
		for (int u = 0; u < N; u++) {
			if (!used[u] && (v == -1 || dist[u] < dist[v])) v = u;
		}
		if (v == -1) break;
		used[v] = true;
		for (int u = 0; u < N; u++) {
			int nd = dist[v] + getCost(v, u);
			if (nd < dist[u] || (nd == dist[u] && v < next[u])) {
				next[u] = v;
				dist[u] = nd;
			}
		}
	}

	int p = S;
	while (p != G) {
		std::cout << p << " ";
		p = next[p];
	}
	std::cout << p << std::endl;
}

int main()
{
	std::cin >> N >> M >> S >> G;
	for (int i = 0; i < M; i++) {
		int a, b, c;
		std::cin >> a >> b >> c;
		set_abc(a, b, c);
	}
	resolve();
}
0