結果

問題 No.748 yuki国のお財布事情
ユーザー poyompypoyompy
提出日時 2018-10-19 21:34:09
言語 C++11
(gcc 11.4.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
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 AC 2 ms
5,248 KB
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 1 ms
5,248 KB
testcase_06 AC 1 ms
5,248 KB
testcase_07 AC 1 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 1 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 16 ms
5,248 KB
testcase_14 AC 28 ms
5,248 KB
testcase_15 AC 19 ms
5,248 KB
testcase_16 AC 51 ms
5,248 KB
testcase_17 AC 103 ms
5,248 KB
testcase_18 AC 121 ms
5,248 KB
testcase_19 AC 132 ms
5,248 KB
testcase_20 AC 117 ms
5,248 KB
testcase_21 AC 114 ms
5,248 KB
testcase_22 AC 2 ms
5,248 KB
testcase_23 AC 2 ms
5,248 KB
testcase_24 AC 1 ms
5,248 KB
testcase_25 AC 93 ms
5,248 KB
testcase_26 AC 111 ms
5,248 KB
testcase_27 AC 109 ms
5,248 KB
testcase_28 AC 91 ms
5,248 KB
権限があれば一括ダウンロードができます

ソースコード

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