結果

問題 No.114 遠い未来
コンテスト
ユーザー ciel
提出日時 2015-10-30 06:01:21
言語 C++11（廃止可能性あり）
(gcc 13.3.0 + boost 1.89.0)
結果
WA  
実行時間 -
コード長 1,833 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 812 ms
コンパイル使用メモリ 71,064 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-09-13 04:48:04
合計ジャッジ時間 9,378 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2 WA * 1
other AC * 5 WA * 19 TLE * 1
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:53:14: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   53 |         scanf("%d%d%d",&N,&M,&T);
      |         ~~~~~^~~~~~~~~~~~~~~~~~~
main.cpp:56:22: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   56 |                 scanf("%d%d%d",&a,&b,&c);
      |                 ~~~~~^~~~~~~~~~~~~~~~~~~
main.cpp:61:34: warning: ignoring return value of ‘int scanf(const char*, ...)’ declared with attribute ‘warn_unused_result’ [-Wunused-result]
   61 |         for(int i=0;i<T;i++)scanf("%d",&v[i]),v[i]--;
      |                             ~~~~~^~~~~~~~~~~~

ソースコード

diff #
raw source code 

#include <cstdio>
#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 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] );

  weight OPT[(1 << numT)][n];
  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;
}

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{
		printf("%d\n",0);
	}
}
0