C++の練習を兼ねて, AtCoder Regular Contest 097 の 問題D (Equals) を解いてみた.
■感想.
1. 問題Dは, 方針が見えなかったので, 解説を参考に実装して, AC版に到達できたので, 良かったと思う.
2. 個人的には, 摩訶不思議な感じがして, 非常に面白いと感じた.
3. Union-Find木 の 復習が出来たので, 非常に良かったと思う.
※ 公式のライブラリを拝借させて頂いてます.
4. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイト AtCoder Regular Contest 097 解説 を ご覧下さい.
■C++版プログラム(問題D／AC版).
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
// 解き直し. // https://img.atcoder.jp/arc097/editorial.pdf // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; #define repex(i, a, b, c) for(int i = a; i < b; i += c) #define repx(i, a, b) repex(i, a, b, 1) #define rep(i, n) repx(i, 0, n) #define repr(i, a, b) for(int i = a; i >= b; i--) int p[101010]; // https://github.com/atcoder/live_library/blob/master/uf.cpp // UnionFind // coding: https://youtu.be/TdR816rqc3s?t=726 // comment: https://youtu.be/TdR816rqc3s?t=6822 // -> 一部改変. struct UnionFind{ vector<int> d; UnionFind(int n = 0): d(n, -1) {} int find(int x){ return (d[x] < 0) ? x : (d[x] = find(d[x])); } bool unite(int x, int y){ x = find(x); y = find(y); if(x == y) return false; if(d[x] > d[y]) swap(x, y); d[x] += d[y]; d[y] = x; return true; } bool same(int x, int y){ return find(x) == find(y); } int size(int x){ return -d[find(x)]; } }; int main(){ // 1. 入力情報. int N, M; scanf("%d %d", &N, &M); rep(i, N){ scanf("%d", &p[i]); p[i]--; } // 2. UnionFind木 に, 辺 を 登録. UnionFind uf(N); rep(i, M){ // 2-1. 辺を取得. int x, y; scanf("%d %d", &x, &y); x--, y--; // 2-2. グラフに辺を登録. uf.unite(x, y); } // 3. i と p[i] が 同じ連結成分にあるか？ int ans = 0; rep(i, N) if(uf.same(i, p[i])) ans++; // 4. 出力. printf("%d\n", ans); return 0; } |
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 |
[入力例] 5 2 5 3 1 4 2 1 3 5 4 [出力例] 2 ※AtCoderテストケースより [入力例] 3 2 3 2 1 1 2 2 3 [出力例] 3 ※AtCoderテストケースより [入力例] 10 8 5 3 6 8 7 10 9 1 2 4 3 1 4 1 5 9 2 5 6 5 3 5 8 9 7 9 [出力例] 8 ※AtCoderテストケースより [入力例] 5 1 1 2 3 4 5 1 5 [出力例] 5 ※AtCoderテストケースより [入力例] 15 10 15 13 9 8 7 3 2 1 6 5 4 12 11 10 14 3 1 4 1 5 2 6 2 6 7 8 9 10 9 11 2 15 9 13 14 [出力例] 3 ※実際, {1, 3, 4}, {2, 5, 6, 7, 11}, {8, 9, 10, 15}, {12}, {13, 14} と 連結成分に分割され, {i, p[i]} = {5, 7}, {7, 2}, {12, 12} の 3つの組み合わせがカウントされることが分かる. |