結果
問題 | No.922 東北きりきざむたん |
ユーザー | startcpp |
提出日時 | 2019-11-08 22:50:02 |
言語 | C++11 (gcc 11.4.0) |
結果 |
AC
|
実行時間 | 394 ms / 2,000 ms |
コード長 | 2,966 bytes |
コンパイル時間 | 1,100 ms |
コンパイル使用メモリ | 65,460 KB |
実行使用メモリ | 35,736 KB |
最終ジャッジ日時 | 2024-09-15 02:00:26 |
合計ジャッジ時間 | 6,544 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 7 ms
24,244 KB |
testcase_01 | AC | 8 ms
25,944 KB |
testcase_02 | AC | 6 ms
21,144 KB |
testcase_03 | AC | 7 ms
21,416 KB |
testcase_04 | AC | 7 ms
21,320 KB |
testcase_05 | AC | 7 ms
21,112 KB |
testcase_06 | AC | 8 ms
26,028 KB |
testcase_07 | AC | 7 ms
21,404 KB |
testcase_08 | AC | 7 ms
21,420 KB |
testcase_09 | AC | 63 ms
26,556 KB |
testcase_10 | AC | 58 ms
25,076 KB |
testcase_11 | AC | 61 ms
27,280 KB |
testcase_12 | AC | 22 ms
26,516 KB |
testcase_13 | AC | 23 ms
22,164 KB |
testcase_14 | AC | 92 ms
27,948 KB |
testcase_15 | AC | 14 ms
25,640 KB |
testcase_16 | AC | 199 ms
29,736 KB |
testcase_17 | AC | 203 ms
29,608 KB |
testcase_18 | AC | 199 ms
29,612 KB |
testcase_19 | AC | 197 ms
29,616 KB |
testcase_20 | AC | 193 ms
29,608 KB |
testcase_21 | AC | 214 ms
29,664 KB |
testcase_22 | AC | 216 ms
29,576 KB |
testcase_23 | AC | 373 ms
30,308 KB |
testcase_24 | AC | 394 ms
30,440 KB |
testcase_25 | AC | 386 ms
30,700 KB |
testcase_26 | AC | 380 ms
30,580 KB |
testcase_27 | AC | 385 ms
30,532 KB |
testcase_28 | AC | 71 ms
26,152 KB |
testcase_29 | AC | 219 ms
35,736 KB |
ソースコード
//考察10分、実装10分、デバッグたくさん (同じ木の間の移動を見落とし、その他実装上の見落とし) #include <iostream> #include <vector> #include <algorithm> #define rep(i, n) for(i = 0; i < n; i++) #define int long long using namespace std; int n, m, q; vector<int> et[100000]; int vcost[100000]; bool used[100000]; int treeSum[100000]; int parent[100000]; int depth[100000]; int dfs(int p, int v, int dep) { int i; used[v] = true; parent[v] = p; depth[v] = dep; int ret = vcost[v]; rep(i, et[v].size()) { int nv = et[v][i]; if (nv == p) continue; ret += dfs(v, nv, dep + 1); } return treeSum[v] = ret; } int calcInitCost(int p, int v) { int i; int ret = 0; rep(i, et[v].size()) { int nv = et[v][i]; if (nv == p) continue; ret += calcInitCost(v, nv); } ret += treeSum[v]; return ret; } int sumcost[100000]; int rt; int dfs2(int p, int v, int sumCost) { int i; sumcost[v] = sumCost; if (p != -1 && et[v].size() == 1) { return sumCost; } int ret = sumCost; rep(i, et[v].size()) { int nv = et[v][i]; if (nv == p) continue; int res = dfs2(v, nv, sumCost + treeSum[rt] - 2 * treeSum[nv]); ret = min(ret, res); } return ret; } // struct UF { int par[100000]; UF() { int i; rep(i, 100000) par[i] = i; } int root(int x) { if (par[x] == x) return x; return par[x] = root(par[x]); } void unit(int x, int y) { x = root(x); y = root(y); par[x] = y; } bool same(int x, int y) { return root(x) == root(y); } }; int parents[20][100000]; UF uf; int lca(int a, int b) { if (depth[a] < depth[b]) { swap(a, b); } int i; int diff = depth[a] - depth[b]; rep(i, 20) { if ((diff >> i) & 1) { a = parents[i][a]; } } for (i = 19; i >= 0; i--) { if (parents[i][a] != parents[i][b]) { a = parents[i][a]; b = parents[i][b]; } } if (a == b) return a; return parent[a]; } signed main() { int i; cin >> n >> m >> q; rep(i, m) { int u, v; cin >> u >> v; u--; v--; et[u].push_back(v); et[v].push_back(u); uf.unit(u, v); } rep(i, n) { if (used[i]) continue; dfs(-1, i, 0); } rep(i, n) parents[0][i] = parent[i]; rep(i, 19) { int j; rep(j, n) { parents[i + 1][j] = parents[i][parents[i][j]]; } } int ans1 = 0; rep(i, q) { int a, b; cin >> a >> b; a--; b--; if (uf.same(a, b)) { int l = lca(a, b); ans1 += depth[a] + depth[b] - 2 * depth[l]; continue; } vcost[a]++; vcost[b]++; } //rep(i, n) { cout << "vcost[" << i + 1 << "] = " << vcost[i] << endl; } rep(i, n) { used[i] = false; } int ans2 = 0; rep(i, n) { if (used[i]) continue; //cout << "i = " << i << endl; dfs(-1, i, 0); int initCost = calcInitCost(-1, i); initCost -= treeSum[i]; rt = i; int res = dfs2(-1, i, initCost); ans2 += res; } /*rep(i, n) { cout << "v = " << i + 1 << ", treeSum = " << treeSum[i] << ", sumcost = " << sumcost[i] << endl; }*/ cout << ans1 + ans2 << endl; return 0; }