結果
問題 | No.748 yuki国のお財布事情 |
ユーザー | ikd |
提出日時 | 2018-10-19 06:45:48 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
RE
|
実行時間 | - |
コード長 | 2,080 bytes |
コンパイル時間 | 982 ms |
コンパイル使用メモリ | 88,936 KB |
実行使用メモリ | 12,240 KB |
最終ジャッジ日時 | 2024-11-14 04:10:46 |
合計ジャッジ時間 | 4,697 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
6,816 KB |
testcase_01 | AC | 2 ms
6,820 KB |
testcase_02 | AC | 2 ms
6,816 KB |
testcase_03 | AC | 2 ms
6,820 KB |
testcase_04 | AC | 2 ms
6,820 KB |
testcase_05 | AC | 2 ms
6,820 KB |
testcase_06 | AC | 2 ms
6,820 KB |
testcase_07 | AC | 2 ms
6,820 KB |
testcase_08 | AC | 1 ms
6,816 KB |
testcase_09 | AC | 2 ms
6,816 KB |
testcase_10 | AC | 2 ms
6,816 KB |
testcase_11 | AC | 2 ms
6,816 KB |
testcase_12 | AC | 2 ms
6,820 KB |
testcase_13 | AC | 15 ms
6,816 KB |
testcase_14 | RE | - |
testcase_15 | AC | 18 ms
6,820 KB |
testcase_16 | AC | 49 ms
6,816 KB |
testcase_17 | AC | 117 ms
9,460 KB |
testcase_18 | AC | 127 ms
8,528 KB |
testcase_19 | RE | - |
testcase_20 | AC | 117 ms
8,144 KB |
testcase_21 | AC | 133 ms
10,448 KB |
testcase_22 | AC | 2 ms
6,820 KB |
testcase_23 | AC | 2 ms
6,816 KB |
testcase_24 | AC | 2 ms
6,820 KB |
testcase_25 | AC | 101 ms
8,656 KB |
testcase_26 | AC | 141 ms
11,792 KB |
testcase_27 | AC | 143 ms
12,240 KB |
testcase_28 | AC | 90 ms
6,820 KB |
ソースコード
#include <algorithm> #include <iostream> #include <numeric> #include <queue> #include <vector> #define rep(i, n) for (int i = 0; i < (int)(n); i++) using namespace std; struct UnionFind { vector<int> par; vector<int> rank; UnionFind(int n) { par.resize(n); rep(i, n) par[i] = i; rank.resize(n, 0); } int find(int i) { if (i == par[i]) return par[i]; else return par[i] = find(par[i]); } void unite(int i, int j) { i = find(i), j = find(j); if (i == j) return; if (rank[i] < rank[j]) { par[i] = j; } else { par[j] = i; if (rank[i] == rank[j]) rank[i]++; } } }; int main() { int n, m, k; cin >> n >> m >> k; struct Edge { int a, b, c; }; vector<Edge> edges; rep(_, m) { int a, b, c; cin >> a >> b >> c; edges.push_back(Edge{a - 1, b - 1, c}); } long long sum = 0; UnionFind uf(n); vector<int> special(m, false); rep(_, k) { int i; cin >> i; uf.unite(edges[i - 1].a, edges[i - 1].b); sum += (long long)edges[i - 1].c; special[i] = true; } struct E { int to, c; bool operator<(const E &r) const { return c > r.c; } }; vector<vector<E>> g(n); rep(i, m) { auto a = uf.find(edges[i].a), b = uf.find(edges[i].b); if (a != b) { g[a].push_back(E{b, edges[i].c}); g[b].push_back(E{a, edges[i].c}); } } auto prim = [&](vector<vector<E>> &g) { vector<int> visited(n, false); priority_queue<E> q; auto s = uf.find(0); visited[s] = true; for (auto e : g[s]) { q.push(e); } long long ret = 0; while (q.size() > 0) { auto e = q.top(); q.pop(); if (visited[e.to]) continue; visited[e.to] = true; ret += (long long)e.c; for (auto ne : g[e.to]) { if (!visited[ne.to]) { q.push(ne); } } } return ret; }; sum += prim(g); auto tot = accumulate( edges.begin(), edges.end(), 0LL, [&](long long res, const auto &e) { return res + (long long)e.c; }); cout << tot - sum << endl; return 0; }