結果

問題 No.1690 Power Grid
ユーザー umezoumezo
提出日時 2021-09-24 23:52:37
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 104 ms / 3,000 ms
コード長 2,256 bytes
コンパイル時間 2,455 ms
コンパイル使用メモリ 215,284 KB
実行使用メモリ 4,384 KB
最終ジャッジ日時 2023-09-18 22:29:13
合計ジャッジ時間 4,231 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
4,384 KB
testcase_01 AC 2 ms
4,380 KB
testcase_02 AC 2 ms
4,376 KB
testcase_03 AC 2 ms
4,380 KB
testcase_04 AC 2 ms
4,376 KB
testcase_05 AC 1 ms
4,380 KB
testcase_06 AC 69 ms
4,376 KB
testcase_07 AC 70 ms
4,380 KB
testcase_08 AC 69 ms
4,376 KB
testcase_09 AC 70 ms
4,380 KB
testcase_10 AC 3 ms
4,376 KB
testcase_11 AC 2 ms
4,376 KB
testcase_12 AC 2 ms
4,380 KB
testcase_13 AC 2 ms
4,376 KB
testcase_14 AC 12 ms
4,376 KB
testcase_15 AC 3 ms
4,376 KB
testcase_16 AC 92 ms
4,380 KB
testcase_17 AC 53 ms
4,376 KB
testcase_18 AC 23 ms
4,376 KB
testcase_19 AC 86 ms
4,380 KB
testcase_20 AC 104 ms
4,380 KB
testcase_21 AC 3 ms
4,380 KB
testcase_22 AC 36 ms
4,384 KB
testcase_23 AC 3 ms
4,376 KB
testcase_24 AC 8 ms
4,376 KB
testcase_25 AC 4 ms
4,376 KB
testcase_26 AC 3 ms
4,380 KB
testcase_27 AC 1 ms
4,376 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define ALL(v) v.begin(),v.end()
typedef long long ll;

#include <bits/stdc++.h>
using namespace std;

const int MAX=20;
const ll INF=1e15;

ll dis[MAX][MAX];
int n,m,k;

void floyd(){
  rep(k,n){
    rep(i,n){
      if(dis[i][k]==INF) continue;
      rep(j,n){
        if(dis[k][j]==INF) continue;
        dis[i][j]=min(dis[i][j],dis[i][k]+dis[k][j]);
      }
    }
  }
}

class Dis{
  public:
  vector<ll> rank,p,siz;
  
  Dis(int s){
    rank.resize(s,0);
    p.resize(s,0);
    siz.resize(s,1);
    rep(i,s) makeSet(i);
  }
  
  void makeSet(int x){
    p[x]=x;
    rank[x]=0;
  }
  
  bool same(int x,int y){
    return findSet(x)==findSet(y);
  }
  
  void unite(int x,int y){
    if(same(x,y)) return;
    link(findSet(x),findSet(y));
  }
  
  void link(int x,int y){
    if(rank[x]>rank[y]){
      p[y]=x;
      siz[x]+=siz[y];
    }
    else{
      p[x]=y;
      siz[y]+=siz[x];
      if(rank[x]==rank[y]) rank[y]++;
    }
  }
  
  int findSet(int x){
    if(x != p[x]) p[x]=findSet(p[x]);
    return p[x];
  }
  
  int size(int x){
    return siz[findSet(x)];
  }
};

//辺e=(u,v)を{w(e),{u,v}}で表す
using Edge=pair<ll,pair<int,int>>;

ll kruskal(int n,vector<Edge> edges){
  sort(ALL(edges));
  ll res=0;
  int m=edges.size();
  Dis dset=Dis(n);
  
  rep(i,m){
    ll w=edges[i].first;
    int u=edges[i].second.first;
    int v=edges[i].second.second;
    
    if(dset.same(u,v)) continue;
    
    res+=w;
    dset.unite(u,v);
  }
  
  return res;
} 

int main(){
  cin>>n>>m>>k;
  
  vector<ll> A(n);
  rep(i,n) cin>>A[i];
  
  rep(i,n){
    rep(j,n){
      if(i==j) dis[i][j]=0;
      else dis[i][j]=INF;
    }
  }
  
  rep(i,m){
    int x,y;
    ll z;
    cin>>x>>y>>z;
    x--,y--;
    dis[x][y]=z;
    dis[y][x]=z;
  }
  
  floyd();
  
  ll mi=INF;
  for(int bit=0;bit<(1<<n);bit++){
    if(__builtin_popcount(bit)!=k) continue;
    vector<int> B;
    ll tmp=0;
    rep(i,n){
      if(bit&(1<<i)){
        B.push_back(i);
        tmp+=A[i];
      }
    }
    vector<Edge> edges;
    for(int i=0;i<k;i++){
      for(int j=i+1;j<k;j++){
        edges.push_back(Edge(dis[B[i]][B[j]],{i,j}));
      }
    }
    tmp+=kruskal(k,edges);
    mi=min(mi,tmp);
  }
  cout<<mi<<endl;
  
  return 0;
}
0