C++の練習を兼ねて, AtCoder Beginner Contest 206 の 問題C (Swappable) ～ 問題D (KAIBUNsyo) を解いてみた.
■感想.
1. 規則性が抽出できたので, AC版に到達できたと思う.
2. 幅優先探索の復習が出来たので, 非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 206 解説 の 各リンク を ご覧下さい.
■C++版プログラム(問題C／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 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using LL = long long; #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--) #define a first #define b second int main(){ // 1. 入力情報. int N; scanf("%d", &N); map<LL, LL> m; rep(i, N){ LL a; scanf("%lld", &a); m[a]++; } // 2. 条件を満たす整数の組をカウント. LL ans = (LL)N * (LL)(N - 1) / 2LL; for(auto &p : m) if(p.b > 1) ans -= p.b * (p.b - 1) / 2; // 3. 出力. printf("%lld\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 |
[入力例] 3 1 7 1 [出力例] 2 ※AtCoderテストケースより [入力例] 10 1 10 100 1000 10000 100000 1000000 10000000 100000000 1000000000 [出力例] 45 ※AtCoderテストケースより [入力例] 20 7 8 1 1 4 9 9 6 8 2 4 1 1 9 5 5 5 3 6 4 [出力例] 173 ※AtCoderテストケースより [入力例] 50 123 106 25 45 57 77 113 48 90 105 70 54 74 68 32 55 89 66 83 56 113 114 87 26 66 109 123 71 28 5 25 109 101 39 46 26 61 122 104 48 85 22 51 26 17 107 22 114 92 13 [出力例] 1214 |
■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 71 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using vi = vector<int>; using vvi = vector<vi>; #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--) #define pb push_back int a[202020]; int main(){ // 1. 入力情報. int N; scanf("%d", &N); rep(i, N){ scanf("%d", &a[i]); a[i]--; } // 2. グラフ作成. vvi G(202020); rep(i, N / 2){ // 2-1. 回文を構成していない要素を, 辺に追加. if(a[i] != a[N - 1 - i]){ G[a[i]].pb(a[N - 1 - i]); G[a[N - 1 - i]].pb(a[i]); } } // 3. グラフを連結成分に分解. // https://ja.wikipedia.org/wiki/幅優先探索 auto bfs = [&](vvi &G, int s, int* c, int l){ // 3-1. 空のキュー. queue<int> q; // 3-2. 連結成分番号を設定. c[s] = l; // 3-3. 探索地点 s をキュー q に追加. q.push(s); while(!q.empty()){ // 3-4. キューから取り出す. int u = q.front(); q.pop(); // 3-5. 取り出した要素を処理. for(auto &e : G[u]){ // 3-6. 訪問済であれば, 処理をスキップ. if(c[e]) continue; if(e != s) c[e] = l, q.push(e); } } return; }; int b[202020], c[202020], idx = 0; rep(i, 202020) b[i] = c[i] = 0; rep(i, 202020) if(!c[i]) bfs(G, i, c, ++idx); // 4. 連結成分の要素数を保存し, カウント. int ans = 0; rep(i, 202020) b[c[i]]++; rep(i, 202020) if(b[i] > 1) ans += b[i] - 1; // 5. 出力. 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 |
[入力例] 8 1 5 3 2 5 2 3 1 [出力例] 2 ※AtCoderテストケースより [入力例] 7 1 2 3 4 1 2 3 [出力例] 1 ※AtCoderテストケースより [入力例] 1 200000 [出力例] 0 ※AtCoderテストケースより [入力例] 7 5 6 8 11 7 13 5 [出力例] 2 [入力例] 15 9 9 7 4 8 3 4 1 7 6 10 8 8 4 7 [出力例] 5 [入力例] 50 35 72 56 75 25 88 107 66 86 13 32 20 4 30 102 15 65 89 78 53 71 110 54 57 107 107 104 64 11 93 56 22 73 21 84 96 108 59 22 38 110 51 39 53 26 20 84 111 74 20 [出力例] 24 |