結果

問題 No.748 yuki国のお財布事情
コンテスト
ユーザー poyompy
提出日時 2018-10-19 21:34:09
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 132 ms / 2,000 ms
コード長 1,028 bytes
コンパイル時間 721 ms
コンパイル使用メモリ 70,388 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-18 20:09:30
合計ジャッジ時間 3,059 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 26
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

struct DSU {
  vector<int> dat;
  
  DSU(int n) : dat(n, -1) {}

  int find(int u) {
    if (dat[u] < 0) return u;
    return dat[u] = find(dat[u]);
  }

  bool unite(int u, int v) {
    u = find(u);
    v = find(v);
    if (u == v) return false;
    if (dat[u] > dat[v]) swap(u, v);
    dat[u] += dat[v];
    dat[v] = u;
    return true;
  }
};


int main() {
  int n, m, k;
  cin >> n >> m >> k;
  vector<pair<int, int>> es(m);
  vector<int> us(m), vs(m);
  long long ans = 0;
  for (int i = 0; i < m; i++) {
    int c;
    cin >> us[i] >> vs[i] >> c;
    es[i] = make_pair(c, i);
  }
  DSU dsu(n + 1);
  vector<bool> can(m, true);
  for (int i = 0; i < k; i++) {
    int e;
    cin >> e;
    e--;
    can[e] = false;
    dsu.unite(us[e], vs[e]);
  }
  sort(es.begin(), es.end());
  for (auto e : es) {
    if (!can[e.second]) continue;
    if (!dsu.unite(us[e.second], vs[e.second])) {
      ans += e.first;
    }
  }
  cout << ans << endl;
}
0