結果

問題 No.748 yuki国のお財布事情
ユーザー poyompypoyompy
提出日時 2018-10-19 21:32:46
言語 C++11
(gcc 11.4.0)
結果
RE  
実行時間 -
コード長 1,024 bytes
コンパイル時間 719 ms
コンパイル使用メモリ 70,516 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-18 20:06:55
合計ジャッジ時間 3,312 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
5,248 KB
testcase_01 AC 2 ms
5,248 KB
testcase_02 RE -
testcase_03 AC 2 ms
5,248 KB
testcase_04 AC 2 ms
5,248 KB
testcase_05 AC 2 ms
5,248 KB
testcase_06 AC 2 ms
5,248 KB
testcase_07 AC 2 ms
5,248 KB
testcase_08 AC 2 ms
5,248 KB
testcase_09 AC 2 ms
5,248 KB
testcase_10 AC 2 ms
5,248 KB
testcase_11 AC 2 ms
5,248 KB
testcase_12 AC 2 ms
5,248 KB
testcase_13 AC 14 ms
5,248 KB
testcase_14 RE -
testcase_15 AC 17 ms
5,248 KB
testcase_16 AC 47 ms
5,248 KB
testcase_17 AC 90 ms
5,248 KB
testcase_18 AC 108 ms
5,248 KB
testcase_19 AC 128 ms
5,248 KB
testcase_20 AC 114 ms
5,248 KB
testcase_21 AC 103 ms
5,248 KB
testcase_22 AC 2 ms
5,248 KB
testcase_23 AC 1 ms
5,248 KB
testcase_24 AC 1 ms
5,248 KB
testcase_25 RE -
testcase_26 AC 102 ms
5,376 KB
testcase_27 AC 99 ms
5,248 KB
testcase_28 AC 81 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);
  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