結果

問題 No.1690 Power Grid
コンテスト
ユーザー TKTYI
提出日時 2021-09-24 22:43:47
言語 C++14
(gcc 13.3.0 + boost 1.87.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
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

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