結果

問題 No.1690 Power Grid
コンテスト
ユーザー square1001
提出日時 2021-09-24 21:34:26
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 141 ms / 3,000 ms
コード長 1,333 bytes
コンパイル時間 662 ms
コンパイル使用メモリ 77,740 KB
実行使用メモリ 6,400 KB
最終ジャッジ日時 2024-07-05 10:09:40
合計ジャッジ時間 3,696 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <vector>
#include <iostream>
#include <algorithm>
using namespace std;
int main() {
	int N, M, K;
	cin >> N >> M >> K;
	vector<long long> A(N);
	for (int i = 0; i < N; ++i) {
		cin >> A[i];
	}
	const long long INF = 1LL << 61;
	vector<vector<long long> > d(N, vector<long long>(N, INF));
	for (int i = 0; i < N; ++i) {
		d[i][i] = 0;
	}
	for (int i = 0; i < M; ++i) {
		int a, b; long long c;
		cin >> a >> b >> c;
		--a, --b;
		d[a][b] = c;
		d[b][a] = c;
	}
	for (int i = 0; i < N; ++i) {
		for (int j = 0; j < N; ++j) {
			for (int k = 0; k < N; ++k) {
				d[j][k] = min(d[j][k], d[j][i] + d[i][k]);
			}
		}
	}
	vector<long long> dp(1 << N, INF);
	dp[0] = 0;
	for (int i = 1; i < (1 << N); ++i) {
		for (int j = 0; j < N; ++j) {
			if (((i >> j) & 1) == 0) {
				continue;
			}
			int sub = i - (1 << j);
			long long cost = dp[sub] + A[j];
			if (sub != 0) {
				long long mindist = INF;
				for (int k = 0; k < N; ++k) {
					if (((sub >> k) & 1) == 1) {
						mindist = min(mindist, d[j][k]);
					}
				}
				cost += mindist;
			}
			dp[i] = min(dp[i], cost);
		}
	}
	vector<int> pop(1 << N);
	for (int i = 1; i < (1 << N); ++i) {
		pop[i] = pop[i >> 1] + (i & 1);
	}
	long long ans = INF;
	for (int i = 0; i < (1 << N); ++i) {
		if (pop[i] == K) {
			ans = min(ans, dp[i]);
		}
	}
	cout << ans << endl;
	return 0;
}
0