結果

問題 No.17 2つの地点に泊まりたい
ユーザー data9824
提出日時 2015-05-21 05:36:37
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
WA  
(最新)
AC  
(最初)
実行時間 -
コード長 1,995 bytes
コンパイル時間 756 ms
コンパイル使用メモリ 74,996 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-06 04:19:57
合計ジャッジ時間 2,850 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 17 WA * 10
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <limits>
#include <algorithm>
#include <queue>

using namespace std;

struct Node {
	Node(int distance, int node) : distance(distance), node(node) {}
	int distance;
	int node;
};

struct DistanceGreater : binary_function < Node, Node, bool > {
	bool operator()(const Node& lhs, const Node& rhs) const {
		return lhs.distance > rhs.distance;
	}
};

int shortest(const vector<vector<int> >& costs, size_t start, size_t end) {
	size_t n = costs.size();
	vector<int> distance(n, numeric_limits<int>::max());
	distance[start] = 0;
	priority_queue<Node, vector<Node>, DistanceGreater> undetermined;
	undetermined.push(Node(distance[start], start));
	vector<bool> determined(n, false);
	while (!undetermined.empty()) {
		Node node = undetermined.top();
		undetermined.pop();
		determined[node.node] = true;
		for (size_t adjNode = 0; adjNode < n; ++adjNode) {
			if (determined[adjNode]) {
				continue;
			}
			int cost = costs[node.node][adjNode];
			if (cost == numeric_limits<int>::max()) {
				continue;
			}
			int newDistance = distance[node.node] + cost;
			if (newDistance < distance[adjNode]) {
				distance[adjNode] = newDistance;
				undetermined.push(Node(newDistance, adjNode));
			}
		}
	}
	return distance[end];
}

int main() {
	int n;
	cin >> n;
	vector<int> s(n);
	for (int i = 0; i < n; ++i) {
		cin >> s[i];
	}
	vector<vector<int> > costs(n, vector<int>(n, numeric_limits<int>::max()));
	int m;
	cin >> m;
	for (int i = 0; i < m; ++i) {
		int a, b, c;
		cin >> a >> b >> c;
		costs[a][b] = c;
		costs[b][a] = c;
	}
	int minCost = numeric_limits<int>::max();
	for (int stay1 = 1; stay1 <= n - 2; ++stay1) {
		for (int stay2 = 1; stay2 <= n - 2; ++stay2) {
			if (stay1 == stay2) {
				continue;
			}
			int cost = s[stay1] + s[stay2];
			cost += shortest(costs, 0, stay1);
			cost += shortest(costs, stay1, stay2);
			cost += shortest(costs, stay2, n - 1);
			minCost = min(minCost, cost);
		}
	}
	cout << minCost << endl;
	return 0;
}
0