結果

問題 No.114 遠い未来
ユーザー krotonkroton
提出日時 2014-12-29 02:02:04
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,214 bytes
コンパイル時間 1,357 ms
コンパイル使用メモリ 163,328 KB
実行使用メモリ 6,992 KB
最終ジャッジ日時 2024-06-13 00:47:06
合計ジャッジ時間 12,580 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 23 ms
6,812 KB
testcase_01 AC 1,156 ms
6,940 KB
testcase_02 AC 151 ms
6,940 KB
testcase_03 AC 34 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 4 ms
6,940 KB
testcase_06 AC 480 ms
6,944 KB
testcase_07 AC 2 ms
6,944 KB
testcase_08 WA -
testcase_09 AC 7 ms
6,940 KB
testcase_10 WA -
testcase_11 AC 330 ms
6,992 KB
testcase_12 AC 934 ms
6,940 KB
testcase_13 AC 3,082 ms
6,944 KB
testcase_14 AC 488 ms
6,944 KB
testcase_15 AC 1,523 ms
6,940 KB
testcase_16 AC 229 ms
6,940 KB
testcase_17 AC 294 ms
6,944 KB
testcase_18 AC 1,022 ms
6,940 KB
testcase_19 AC 112 ms
6,944 KB
testcase_20 AC 119 ms
6,944 KB
testcase_21 AC 8 ms
6,940 KB
testcase_22 AC 12 ms
6,944 KB
testcase_23 AC 1 ms
6,940 KB
testcase_24 AC 3 ms
6,944 KB
testcase_25 AC 2 ms
6,940 KB
testcase_26 AC 2 ms
6,940 KB
testcase_27 AC 1 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
 
const int INF = 1 << 26;
struct Edge {
	int u, v, c;
	bool operator<(const Edge& other) const {
		return c < other.c;
	}
};
 
int N, M, T, ts[50];
Edge E[777];
int dist[50][50];
 
int root[50];
inline int find(int x){
	if(x == root[x])return x;
	return root[x] = find(root[x]);
}
inline bool conn(int x, int y){
	x = find(x);
	y = find(y);
	if(x == y)return false;
 
	root[x] = y;
	return true;
}
 
 
int rest_v[50];
int use[50];
inline int min_spanning_tree(){
	for(int i=0;i<N;i++)root[i] = i;
 
	int res = 0;
	for(int i=0;i<M;i++){
		int u = E[i].u, v = E[i].v, c = E[i].c;
		if(!use[u] || !use[v])continue;
		if(conn(u, v)){
			res += c;
		}
	}
 
	for(int i=0;i<T;i++){
		if(find(ts[i]) != find(ts[0]))return INF;
	}
 
	return res;
}
 
int dp[1<<15][50];
int min_steiner_tree(){
  for(int i=0;i<N;i++)for(int j=0;j<N;j++)dist[i][j] = INF;
  for(int i=0;i<N;i++)dist[i][i] = 0;
  for(int i=0;i<M;i++){
  	int u = E[i].u, v = E[i].v, c = E[i].c;
  	dist[u][v] = c;
  	dist[v][u] = c;
  }
 
  if(T < 2)return 0;
  for(int i=0;i<T;i++)for(int j=0;j<N;j++){
  	dp[1 << i][j] = dist[ts[i]][j];
  }
 
  for(int i=1;i<1<<T;i++)if(((i - 1) & i) != 0){
  	for(int j=0;j<N;j++){
  		dp[i][j] = INF;
  		for(int k=(i-1)&i; k > 0; k = (k - 1) & i){
  			dp[i][j] = min(dp[i][j], dp[k][j] + dp[i^k][j]);
  		}
  	}
  	for(int j=0;j<N;j++){
  		for(int k=0;k<N;k++){
  			dp[i][j] = min(dp[i][j], dp[i][k] + dist[k][j]);
  		}
  	}
  }
 
  return dp[(1 << T)-1][ts[0]];
}
 
int main(){
	cin >> N >> M >> T;
	for(int i=0;i<M;i++){
		int u, v, c;
		cin >> u >> v >> c;
		E[i] = Edge{u - 1, v - 1, c};
	}
	for(int i=0;i<T;i++)cin >> ts[i], ts[i]--;
 
	int res;
	if(T <= 14){
		res = min_steiner_tree();
	} else {
		for(int i=0;i<T;i++)use[ts[i]] = true;
 
		int rest = 0;
		for(int i=0;i<N;i++)if(!use[i]){
			rest_v[rest++] = i;
		}
 
		sort(E, E + M);
		res = INF;
 
		for(int mask=0;mask<1<<rest;mask++){
			for(int i=0;i<rest;i++){
				if((mask >> i) & 1){
					use[rest_v[i]] = true;
				}
			}
 
			res = min(res, min_spanning_tree());
 
			for(int i=0;i<rest;i++){
				if((mask >> i) & 1){
					use[rest_v[i]] = false;
				}
			}
		}
	}
 
	cout << res << endl;
	return 0;
}
0