結果

問題 No.1690 Power Grid
ユーザー TKTYITKTYI
提出日時 2021-09-24 22:43:47
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 116 ms / 3,000 ms
コード長 1,563 bytes
コンパイル時間 4,705 ms
コンパイル使用メモリ 245,816 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-07-05 11:03:13
合計ジャッジ時間 5,884 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,944 KB
testcase_05 AC 2 ms
6,944 KB
testcase_06 AC 66 ms
6,940 KB
testcase_07 AC 67 ms
6,940 KB
testcase_08 AC 65 ms
6,944 KB
testcase_09 AC 64 ms
6,944 KB
testcase_10 AC 2 ms
6,940 KB
testcase_11 AC 2 ms
6,940 KB
testcase_12 AC 2 ms
6,944 KB
testcase_13 AC 2 ms
6,944 KB
testcase_14 AC 13 ms
6,940 KB
testcase_15 AC 2 ms
6,940 KB
testcase_16 AC 102 ms
6,944 KB
testcase_17 AC 59 ms
6,940 KB
testcase_18 AC 24 ms
6,940 KB
testcase_19 AC 102 ms
6,944 KB
testcase_20 AC 116 ms
6,940 KB
testcase_21 AC 2 ms
6,940 KB
testcase_22 AC 42 ms
6,940 KB
testcase_23 AC 2 ms
6,940 KB
testcase_24 AC 8 ms
6,940 KB
testcase_25 AC 3 ms
6,944 KB
testcase_26 AC 2 ms
6,944 KB
testcase_27 AC 2 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
#include <atcoder/all>
using namespace std;
using namespace atcoder;
typedef long long int ll;
typedef long double ld;
#define FOR(i,l,r) for(ll i=l;i<r;i++)
#define REP(i,n) FOR(i,0,n)
#define RFOR(i,l,r) for(ll i=r-1;i>=l;i--)
#define RREP(i,n) RFOR(i,0,n)
#define ALL(x) x.begin(),x.end()
#define P pair<ll,ll>
#define F first
#define S second
#define BS(A,x) binary_search(ALL(A),x)
#define LB(A,x) (ll)(lower_bound(ALL(A),x)-A.begin())
#define UB(A,x) (ll)(upper_bound(ALL(A),x)-A.begin())
#define COU(A,x) UB(A,x)-LB(A,x)
template<typename T>using min_priority_queue=priority_queue<T,vector<T>,greater<T>>;
using mint1=modint1000000007;
using mint2=modint998244353;
struct edge{ll u,v,cost;};
bool comp(edge&e1,edge&e2){return e1.cost<e2.cost;}
signed main(){
  ll N,M,K,ans=1e18;cin>>N>>M>>K;
  vector<ll>A(N);REP(i,N)cin>>A[i];
  vector<edge>E(M);
  REP(i,M)cin>>E[i].u>>E[i].v>>E[i].cost;
  REP(i,M)E[i].u--,E[i].v--;
  sort(ALL(E),comp);
  vector<vector<ll>>D(N,vector<ll>(N,1e18));
  REP(i,N)D[i][i]=0;for(auto e:E){D[e.u][e.v]=e.cost;D[e.v][e.u]=e.cost;}
  REP(k,N)REP(i,N)REP(j,N)D[i][j]=min(D[i][j],D[i][k]+D[k][j]);
  vector<bool>X(N,0);REP(i,K)X[i]=1;sort(ALL(X));
  do{
    ll res=0;vector<ll>T;REP(i,N)if(X[i]){res+=A[i];T.push_back(i);}
    dsu Ds(K);vector<edge>ed;
    REP(i,T.size())REP(j,i)ed.push_back(edge{i,j,D[T[i]][T[j]]});
    sort(ALL(ed),comp);
    for(auto e:ed)if(!Ds.same(e.u,e.v)){Ds.merge(e.u,e.v);res+=e.cost;}
    ans=min(ans,res);
  }while(next_permutation(ALL(X)));
  cout<<ans<<endl;
  return 0;
}
0