結果

問題 No.114 遠い未来
ユーザー krotonkroton
提出日時 2014-12-29 02:07:30
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 3,486 ms / 5,000 ms
コード長 2,323 bytes
コンパイル時間 1,568 ms
コンパイル使用メモリ 150,568 KB
実行使用メモリ 7,444 KB
最終ジャッジ日時 2023-09-03 19:56:52
合計ジャッジ時間 14,229 ms
ジャッジサーバーID
(参考情報)
judge12 / judge13
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 25 ms
4,380 KB
testcase_01 AC 1,261 ms
4,376 KB
testcase_02 AC 175 ms
4,380 KB
testcase_03 AC 38 ms
4,380 KB
testcase_04 AC 1 ms
4,380 KB
testcase_05 AC 5 ms
4,384 KB
testcase_06 AC 526 ms
4,380 KB
testcase_07 AC 1 ms
4,380 KB
testcase_08 AC 1 ms
4,376 KB
testcase_09 AC 7 ms
4,380 KB
testcase_10 AC 120 ms
5,104 KB
testcase_11 AC 365 ms
7,444 KB
testcase_12 AC 1,026 ms
4,380 KB
testcase_13 AC 3,486 ms
4,376 KB
testcase_14 AC 535 ms
4,380 KB
testcase_15 AC 1,751 ms
4,380 KB
testcase_16 AC 261 ms
4,384 KB
testcase_17 AC 331 ms
4,376 KB
testcase_18 AC 1,178 ms
4,380 KB
testcase_19 AC 126 ms
4,380 KB
testcase_20 AC 137 ms
4,380 KB
testcase_21 AC 9 ms
4,380 KB
testcase_22 AC 13 ms
4,380 KB
testcase_23 AC 2 ms
4,380 KB
testcase_24 AC 3 ms
4,380 KB
testcase_25 AC 2 ms
4,380 KB
testcase_26 AC 2 ms
4,380 KB
testcase_27 AC 1 ms
4,384 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;
  }

  for(int k=0;k<N;k++)for(int i=0;i<N;i++)for(int j=0;j<N;j++){
  	dist[i][j] = min(dist[i][j], dist[i][k] + dist[k][j]);
  }
  

  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