結果

問題 No.114 遠い未来
ユーザー 古寺いろは古寺いろは
提出日時 2015-03-25 09:37:35
言語 C++11
(gcc 11.4.0)
結果
WA  
実行時間 -
コード長 2,884 bytes
コンパイル時間 1,969 ms
コンパイル使用メモリ 172,712 KB
実行使用メモリ 14,804 KB
最終ジャッジ日時 2024-06-29 00:47:12
合計ジャッジ時間 8,743 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 TLE -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include "bits/stdc++.h"
using namespace std;

#define border 15

int dist[36][36];
int need[36];
int check[36];
int ans;

int uni[36];

int getuni(int a){
	if (uni[a] < 0) return a;
	else return uni[a] = getuni(uni[a]);
}

int connect(int a, int b){
	a = getuni(a);
	b = getuni(b);
	if (a == b) return 0;
	if (uni[a] > uni[b]) swap(a, b);
	uni[a] += uni[b];
	uni[b] = a;
	return 1;
}

int spanningTree(int N){
	for (int i = 0; i < N; i++)
	{
		uni[i] = -1;
	}

	vector<tuple<int, int, int>> v;
	for (int i = 0; i < N; i++)
	{
		if (!check[i]) continue;
		for (int j = i + 1; j < N; j++)
		{
			if (!check[j]) continue;
			v.push_back(make_tuple(dist[i][j], i, j));
		}
	}

	sort(v.begin(), v.end());

	int cost = 0;
	int count = 0;
	for (auto now : v){
		if (connect(get<1>(now), get<2>(now))){
			cost += get<0>(now);
			count++;
		}
	}
	ans = min(ans, cost);
	return cost;
}

int OPT[(1 << border)][35];

int minimum_steiner_tree(const vector<int>& T, int N) {
	int n = N;
	int numT = T.size();
	if (numT <= 1) return 0;

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

	for (int p = 0; p < numT; ++p) // trivial case
	for (int q = 0; q < n; ++q)
		OPT[1 << p][q] = dist[T[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 = 0; E < S; ++E)
		if ((E | S) == 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 ans2 = 999999;
	for (int S = 0; S < (1 << numT); ++S)
	for (int q = 0; q < n; ++q)
		ans2 = min(ans2, OPT[S][q] + OPT[((1 << numT) - 1) - S][q]);

	ans = ans2;
	return ans;
}



int main() {
	int N, M, T;
	cin >> N >> M >> T;
	int MAX = 999999;
	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			dist[i][j] = MAX;
		}
		dist[i][i] = 0;
	}

	for (int i = 0; i < M; i++)
	{
		int a, b, c;
		cin >> a >> b >> c;
		a--; b--;
		dist[a][b] = c;
		dist[b][a] = 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]);
			}
		}
	}

	for (int i = 0; i < N; i++)
	{
		for (int j = 0; j < N; j++)
		{
			cout << dist[i][j] << " ";
		}
		cout << endl;
	}

	vector<int> T2;
	for (int i = 0; i < T; i++)
	{
		int temp;
		cin >> temp;
		temp--;
		T2.push_back(temp);
		need[temp] = 1;
		check[temp] = 1;
	}

	ans = 999999;


	if (T <= border){
		ans = minimum_steiner_tree(T2, N);
	}
	else{
		vector<int> nokori;
		for (int i = 0; i < N; i++)
		{
			if (!need[i]) nokori.push_back(i);
		}

		for (int i = 0; i < (1 << nokori.size()); i++)
		{
			for (int j = 0; j < nokori.size(); j++)
			{
				check[nokori[j]] = (i >> j) % 2;
			}
			int temp = spanningTree(N);
		}
	}
	cout << ans << endl;
}
0