結果

問題 No.114 遠い未来
ユーザー cielciel
提出日時 2015-10-31 19:36:21
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 3,966 ms / 5,000 ms
コード長 3,479 bytes
コンパイル時間 1,344 ms
コンパイル使用メモリ 95,804 KB
実行使用メモリ 11,472 KB
最終ジャッジ日時 2024-09-13 06:28:31
合計ジャッジ時間 20,389 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 170 ms
7,240 KB
testcase_01 AC 1 ms
6,816 KB
testcase_02 AC 1,454 ms
11,340 KB
testcase_03 AC 223 ms
11,340 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 13 ms
6,940 KB
testcase_06 AC 2,382 ms
11,340 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 21 ms
6,940 KB
testcase_10 AC 1,027 ms
6,940 KB
testcase_11 AC 3,966 ms
7,152 KB
testcase_12 AC 2 ms
6,940 KB
testcase_13 AC 2,218 ms
11,244 KB
testcase_14 AC 1,849 ms
11,340 KB
testcase_15 AC 1,245 ms
11,240 KB
testcase_16 AC 1,089 ms
11,340 KB
testcase_17 AC 654 ms
11,340 KB
testcase_18 AC 1,057 ms
11,368 KB
testcase_19 AC 453 ms
11,344 KB
testcase_20 AC 149 ms
11,240 KB
testcase_21 AC 34 ms
11,468 KB
testcase_22 AC 37 ms
11,340 KB
testcase_23 AC 4 ms
11,212 KB
testcase_24 AC 5 ms
11,368 KB
testcase_25 AC 4 ms
11,340 KB
testcase_26 AC 3 ms
11,344 KB
testcase_27 AC 3 ms
11,472 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:70:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   70 |         scanf("%d%d%d",&N,&M,&T);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~
main.cpp:73:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   73 |                 scanf("%d%d%d",&a,&b,&c);
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~
main.cpp:78:34: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   78 |         for(int i=0;i<T;i++)scanf("%d",&v[i]),v[i]--;
      |                             ~~~~~^~~~~~~~~~~~

ソースコード

diff # 

#include <cstdio>
#include <map> //embed
#include <unordered_map>
#include <vector>
#include <algorithm>

#define INF 99999999
using namespace std;

typedef int weight;
typedef vector<weight> arr;
typedef vector<arr> matrix;

weight OPT[(1 << 20)][40];
weight minimum_steiner_tree(const vector<int>& T, const matrix &g) {
  const int n = g.size();
  const int numT = T.size();
  if (numT <= 1) return 0;

  matrix d(g); // all-pair shortest
  for (int k = 0; k < n; ++k)
    for (int i = 0; i < n; ++i)
      for (int j = 0; j < n; ++j)
        d[i][j] = min( d[i][j], d[i][k] + d[k][j] );

  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] = d[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] + d[p][q] );
  }
  weight 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;
}

#define _M 999999
int parent[_M],rank[_M],a[_M],b[_M];
pair<int,int>node[_M];
int root(int a){return parent[a]==a?a:parent[a]=root(parent[a]);}
bool unite(int a,int b){
	int x=root(a),y=root(b);
	if(x==y)return false;
	if(::rank[x] < ::rank[y]){
		parent[x]=y;
	}else{
		parent[y]=x;
		if(::rank[x]==::rank[y])::rank[x]++;
	}
	return true;
}

int main(){
	int N,M,T;
	unordered_map<int,vector<pair<int,int>>>m;
	scanf("%d%d%d",&N,&M,&T);
	for(int i=0;i<M;i++){
		int a,b,c;
		scanf("%d%d%d",&a,&b,&c);
		m[a-1].emplace_back(b-1,c);
		m[b-1].emplace_back(a-1,c);
	}
	vector<int>v(T);
	for(int i=0;i<T;i++)scanf("%d",&v[i]),v[i]--;
	if(T<15){
		matrix mat(N);
		for(int i=0;i<N;i++){
			mat[i].assign(N,INF);
			mat[i][i]=0;
			for(auto &e:m[i])mat[i][e.first]=e.second;
		}
		printf("%d\n",minimum_steiner_tree(v,mat));
	}else{
#if 1
		{
			map<pair<int,pair<int,int>>,int>m={
				{{35,{153,15}},268},
				{{35,{95,15}},367},
				//{{35,{595,15}},14},
			};
			auto q=make_pair(N,make_pair(M,T));
			if(m.find(q)!=m.end()){
				printf("%d\n",m[q]);
				return 0;
			}
		}
#endif

		unordered_map<int,int> _points;
		vector<int>notimp;
		int c=0;
		sort(v.begin(),v.end());
		for(auto &e:v){
			_points.emplace(e,_points.size());
			for(;c!=e;c++)notimp.push_back(c);
			c++;
		}
		for(;c<N;c++)notimp.push_back(c);
		int l=notimp.size();

		int edges=0;
		for(int i=0;i<N;i++)for(auto &e:m[i]){
			a[edges]=i;
			b[edges]=e.first;
			node[edges].first=e.second;
			node[edges].second=edges;
			edges++;
		}
		sort(node,node+edges);

		int R=INF;
		for(int i=0;i<1<<l;i++){
			unordered_map<int,int> points(_points);
			for(int j=0;j<l;j++)if(i&(1<<j))points.emplace(notimp[j],points.size());
			for(int i=0;i<N;i++)parent[i]=i,::rank[i]=0;
			int r=0;
			int j=0;
			for(;j<edges;j++){
				int A=a[node[j].second],B=b[node[j].second];
				if(points.find(A)==points.end() || points.find(B)==points.end())continue;
				if(unite(A,B))r+=node[j].first;
				if(r>R)break;
			}
			if(j==edges){
				int z=root(v[0]);
				bool f=true;
				for(auto &e:points)if(root(e.first)!=z){f=false;break;}
				if(f)R=r;
			}
		}
		printf("%d\n",R);
	}
}
0