結果

問題 No.114 遠い未来
ユーザー krotonkroton
提出日時 2014-12-29 01:30:07
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 579 ms / 5,000 ms
コード長 2,600 bytes
コンパイル時間 1,601 ms
コンパイル使用メモリ 163,460 KB
実行使用メモリ 6,656 KB
最終ジャッジ日時 2024-06-13 00:36:19
合計ジャッジ時間 5,882 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 26 ms
5,248 KB
testcase_01 AC 579 ms
5,376 KB
testcase_02 AC 181 ms
5,376 KB
testcase_03 AC 28 ms
5,376 KB
testcase_04 AC 1 ms
5,376 KB
testcase_05 AC 4 ms
5,376 KB
testcase_06 AC 349 ms
5,376 KB
testcase_07 AC 2 ms
5,376 KB
testcase_08 AC 2 ms
5,376 KB
testcase_09 AC 7 ms
5,376 KB
testcase_10 AC 121 ms
5,376 KB
testcase_11 AC 364 ms
6,656 KB
testcase_12 AC 486 ms
5,376 KB
testcase_13 AC 298 ms
5,376 KB
testcase_14 AC 229 ms
5,376 KB
testcase_15 AC 166 ms
5,376 KB
testcase_16 AC 152 ms
5,376 KB
testcase_17 AC 84 ms
5,376 KB
testcase_18 AC 116 ms
5,376 KB
testcase_19 AC 55 ms
5,376 KB
testcase_20 AC 16 ms
5,376 KB
testcase_21 AC 5 ms
5,376 KB
testcase_22 AC 5 ms
5,376 KB
testcase_23 AC 2 ms
5,376 KB
testcase_24 AC 2 ms
5,376 KB
testcase_25 AC 2 ms
5,376 KB
testcase_26 AC 2 ms
5,376 KB
testcase_27 AC 2 ms
5,376 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(int now){
	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;
			if(res >= now)break;
		}
	}

	if(res >= now)return INF;

	for(int i=0;i<T;i++){
		if(find(ts[i]) != find(ts[0]))return INF;
	}

	return res;
}

int OPT[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;
  }

  const int n = N;
  const int numT = T;
  if (numT <= 1) return 0;

  for (int S = 0; S < (1 << numT); ++S)
    for (int x = 0; x < n; ++x)
      OPT[S][x] = INF;

  for (int p = 0; p < numT; ++p) // trivial case
    for (int q = 0; q < n; ++q)
      OPT[1 << p][q] = dist[ts[p]][q];

  for (int S = 1; S < (1 << numT); ++S) { // DP step
    if (!(S & (S-1))) continue;
    for (int p = 0; p < n; ++p)
      for (int E = (S-1)&S; E ; E = (E-1)&S)
          OPT[S][p] = min( OPT[S][p], OPT[E][p] + OPT[S-E][p] );
    for (int p = 0; p < n; ++p)
      for (int q = 0; q < n; ++q)
        OPT[S][p] = min( OPT[S][p], OPT[S][q] + dist[p][q] );
  }
  int ans = INF;
  for (int S = 0; S < (1 << numT); ++S)
    for (int q = 0; q < n; ++q)
      ans = min(ans, OPT[S][q] + OPT[((1 << numT)-1)-S][q]);
  return ans;
}

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(res));

			for(int i=0;i<rest;i++){
				if((mask >> i) & 1){
					use[rest_v[i]] = false;
				}
			}
		}
	}

	cout << res << endl;
	return 0;
}
0