結果

問題 No.114 遠い未来
ユーザー atetubouatetubou
提出日時 2014-11-06 16:32:26
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 2,058 ms / 5,000 ms
コード長 3,384 bytes
コンパイル時間 1,655 ms
コンパイル使用メモリ 162,692 KB
実行使用メモリ 13,884 KB
最終ジャッジ日時 2023-08-29 00:49:30
合計ジャッジ時間 15,254 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 15 ms
4,380 KB
testcase_01 AC 667 ms
9,508 KB
testcase_02 AC 2,051 ms
13,576 KB
testcase_03 AC 34 ms
4,380 KB
testcase_04 AC 1 ms
4,376 KB
testcase_05 AC 3 ms
4,376 KB
testcase_06 AC 2,058 ms
13,884 KB
testcase_07 AC 2 ms
4,376 KB
testcase_08 AC 1 ms
4,380 KB
testcase_09 AC 6 ms
4,380 KB
testcase_10 AC 80 ms
4,660 KB
testcase_11 AC 234 ms
7,584 KB
testcase_12 AC 687 ms
9,500 KB
testcase_13 AC 679 ms
9,544 KB
testcase_14 AC 2,035 ms
13,620 KB
testcase_15 AC 2,039 ms
13,656 KB
testcase_16 AC 165 ms
4,380 KB
testcase_17 AC 104 ms
4,376 KB
testcase_18 AC 133 ms
4,376 KB
testcase_19 AC 67 ms
4,376 KB
testcase_20 AC 19 ms
4,380 KB
testcase_21 AC 5 ms
4,380 KB
testcase_22 AC 7 ms
4,376 KB
testcase_23 AC 2 ms
4,380 KB
testcase_24 AC 2 ms
4,380 KB
testcase_25 AC 2 ms
4,376 KB
testcase_26 AC 2 ms
4,380 KB
testcase_27 AC 2 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

#define rep(i,n) for(int i=0;i<(int)(n);++i)
#define iter(a) __typeof(a.begin())
#define FOR(it,a) for(iter(a)it=a.begin();it!=a.end();++it)
#define F first
#define S second
#define SZ(a) (int)((a).size())
#define sz(a) SZ(a)
#define pb(a) push_back(a)
#define mp(a,b) make_pair(a,b)
#define ALL(a) (a).begin(),(a).end()
using namespace std;



typedef long long ll;
typedef pair<int,int> PI;

#define PR(...) do{cerr << "line : " << __LINE__ << endl; pr(#__VA_ARGS__, __VA_ARGS__);}while(0);
template<class T>
void pr(const string& name, T t){
  cerr << name << ": " << t << endl;
}

template<typename T, typename ... Types>
void pr(const string& names, T t, Types ... rest) {
  auto comma_pos = names.find(',');
  cerr << names.substr(0, comma_pos) << ": " << t << ", ";
  
  auto next_name_pos = names.find_first_not_of(" \t\n", comma_pos + 1);
  pr(string(names, next_name_pos), rest ...);
}

template<class T,class U> ostream& operator<< (ostream& o, const pair<T,U>& v){return o << "(" << v.F << ", " << v.S << ")";}
template<class T> ostream& operator<< (ostream& o, const vector<T>& v){o << "{";rep(i,SZ(v)) o << (i?", ":"") << v[i];return o << "}";}


const int MAX_N=40;
int cost[MAX_N][MAX_N];
const int THRESHOLD=16;

int dp[1<<THRESHOLD][MAX_N];
int un[MAX_N];
int find(int x){
  if(x == un[x]) return x;
  return un[x] = find(un[x]);
}

void unit(int a,int b){
  un[find(a)] = find(b);
}

int main(int argc, char *argv[])
{
  int n,m,t;
  cin >> n >> m >> t;
  rep(i,n)rep(j,n) cost[i][j] = 1e9;
  vector<pair<int,PI > > ed;
  rep(i,m){
    int a,b,c;
    cin >> a >> b >> c;
    --a,--b;
    ed.pb(mp(c,mp(a,b)));
    cost[b][a] = cost[a][b] = min(cost[a][b],c);
  }
  rep(i,n) cost[i][i] = 0;
  rep(k,n)rep(i,n)rep(j,n)
    cost[i][j] = min(cost[i][j], cost[i][k]+cost[k][j]);
  
  vector<int> terminal(t);
  rep(i, t){
    cin >> terminal[i];
    --terminal[i];
  }
  
  if(t>THRESHOLD){
    sort(ALL(ed));
    vector<int> nont;
    sort(ALL(terminal));
    rep(i,n)
      if(!binary_search(ALL(terminal),i))
        nont.pb(i);
    //cout << nont << endl;
    int ans = 1e9;
    rep(i,1<<(n-t)){
      vector<bool> use(n);
      rep(j,n-t) use[nont[j]] = (i>>j)&1;
      rep(j,t) use[terminal[j]] = true;
      rep(j,n) un[j] = j;
      int cc = 0;
      for(const auto& e : ed){
        if(!use[e.S.F] || !use[e.S.S]) continue;
        if(find(e.S.F) == find(e.S.S)) continue;
        unit(e.S.F,e.S.S);
        cc += e.F;
        if(cc >= ans) break;
      }
      if(cc >= ans) continue;
      bool ok = true;
      for(auto t : terminal){
        if(find(t) == find(terminal[0])) continue;
        ok = false;
        break;
      }
      
      if(ok){
        //cout << "hoge" << endl;
        ans = cc;
      }
    }
    
    cout << ans << endl;
  }else{
    rep(S,1<<t)rep(i,n) dp[S][i] = 2e9;
    rep(i,t)rep(j,n) dp[1<<i][j] = cost[terminal[i]][j];

    for(int S = 3; S < (1<<t); ++S){
      int sub = (S-1) & S;
      do{
        rep(i,n) dp[S][i] = min(dp[S][i], dp[sub][i] + dp[sub^S][i]);
        sub = (sub-1) & S;
      }while(sub);
      rep(p,n)rep(q,n)
        dp[S][p] = min(dp[S][p], dp[S][q]+cost[q][p]);
    }
    
    cout << dp[(1<<t)-1][terminal[0]] << endl;
  }
  
  return 0;
}
0