C++の練習を兼ねて, AtCoder Beginner Contest 282 の 問題C (String Delimiter) ~ 問題D (Make Bipartite 2) を解いてみた.
■感想.
1. 問題C, D は, 方針を絞り込めたので, AC版に到達できたと思う.
2. 個人的には, 問題D で, 幅優先探索 の 復習が出来たので, 非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 282 解説 の 各リンク を ご覧下さい.
■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 33 34 |
// 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--) #define pb push_back int main(){ // 1. 入力情報. int N; char c[202020]; scanf("%d %s", &N, c); string S(c); // 2. 文字列作成. string ans; bool f = false; rep(i, N){ if(S[i] == ','){ ans.pb(f ? ',' : '.'); continue; } if(S[i] == '"') f = !f; ans.pb(S[i]); } // 3. 出力. printf("%s\n", ans.c_str()); 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 |
[入力例] 8 "a,b"c,d [出力例] "a,b"c.d ※AtCoderテストケースより [入力例] 5 ,,,,, [出力例] ..... ※AtCoderテストケースより [入力例] 20 a,"t,"c,"o,"d,"e,"r, [出力例] a."t,"c."o,"d."e,"r. ※AtCoderテストケースより [入力例] 45 "s,t,"r,"i,n,"g,"d","e,""l,i,m",i,"""t,e,"r"" [出力例] "s,t,"r."i,n,"g."d"."e,""l,i,m".i."""t,e,"r"" [入力例] 70 ""a,","aa,","aaa,","aaaa,","aaaaa,","aaaaaa,","aaaaaaa,","aaaaaaaa,"," [出力例] ""a.","aa.","aaa.","aaaa.","aaaaa.","aaaaaa.","aaaaaaa.","aaaaaaaa."," |
■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 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 |
// C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using LL = long long; 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 #define a first #define b second int main(){ // 1. 入力情報. int N, M; scanf("%d %d", &N, &M); vvi G(N); rep(i, M){ int u, v; scanf("%d %d", &u, &v); --u; --v; G[u].pb(v); G[v].pb(u); } // 2. bfs. // https://ja.wikipedia.org/wiki/幅優先探索 auto bfs = [&](vvi &G, int s, int* c, int l){ // 空のキュー. queue<int> q; // 連結成分番号を設定. c[s] = l; // 探索地点 s をキュー q に追加. q.push(s); while(!q.empty()){ // キューから取り出す. int u = q.front(); q.pop(); // 取り出した要素を処理. for(auto &e : G[u]) if(!c[e] && e != s) c[e] = l, q.push(e); } }; int c[N], n = 0; rep(i, N) c[i] = 0; rep(i, N) if(!c[i]) bfs(G, i, c, ++n); // 3. 連結成分ごとのグラフ. vvi CG(n + 1); rep(i, N) CG[c[i] - 1].pb(i); // 4. 二部グラフ判定・集計. // -> 各連結成分が, 二部グラフとなる必要があるはず. bool isBipartite = true; LL ans = 0; rep(i, n){ // 4-1. 座標圧縮. map<int, int> m; set<int> st; for(auto &p : CG[i]) m[p] = 0, st.insert(p); int idx = -1; for(auto &p : m) p.b = ++idx; // 4-2. 最短距離. // -> 頂点番号 を 座標圧縮したので, bfsは, ここに埋め込んだ. int d[idx + 1]; rep(j, idx + 1) d[j] = -1; queue<int> q; int s = CG[i][0]; d[m[s]] = 0; q.push(s); while(!q.empty()){ int u = q.front(); q.pop(); for(auto &v : G[u]) if(st.count(v) && d[m[v]] == -1 && v != s) d[m[v]] = d[m[u]] + 1, q.push(v); } // 4-3. 距離の偶奇. LL e = 0, o = 0; rep(j, idx + 1){ if(d[j] & 1) ++o; else ++e; } // 4-4. 二部グラフか? LL t = 0; for(auto &p : CG[i]){ LL c = 0; for(auto &q : G[p]){ // 距離の偶奇が同じ場合は, 矛盾. if((d[m[q]] & 1) == (d[m[p]] & 1)) isBipartite = false; // 距離の偶奇が異なるものをカウント. if((d[m[q]] & 1) != (d[m[p]] & 1)) ++c; } // 隣接頂点どうしは, 辺が存在しているので除外. t += (d[m[p]] & 1) ? (e - c) : (o - c); } // 4-5. 加算. ans += (LL)(idx + 1) * (LL)(N - idx - 1); ans += t; } // 5. 出力. printf("%lld\n", isBipartite ? ans / 2 : 0); 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 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 |
[入力例] 5 4 4 2 3 1 5 2 3 2 [出力例] 2 ※AtCoderテストケースより [入力例] 4 3 3 1 3 2 1 2 [出力例] 0 ※AtCoderテストケースより [入力例] 9 11 4 9 9 1 8 2 8 3 9 2 8 4 6 7 4 6 7 5 4 5 7 8 [出力例] 9 ※AtCoderテストケースより [入力例] 10 10 1 2 8 1 2 3 9 3 9 10 3 4 5 4 5 6 7 6 5 8 [出力例] 15 [入力例] 19 18 1 2 1 3 1 7 1 8 7 15 8 15 2 4 5 3 4 6 6 5 6 12 13 6 14 12 14 13 14 16 14 17 18 16 19 17 [出力例] 97 [入力例] 12 8 5 6 6 8 5 7 7 8 9 10 10 11 12 11 12 9 [出力例] 54 [入力例] 23 18 1 2 1 9 3 2 11 3 10 9 11 10 13 14 15 14 13 12 15 12 16 18 16 17 17 20 16 21 22 21 19 18 23 19 22 18 [出力例] 215 [入力例] 30 27 4 5 7 5 6 5 8 9 8 11 11 10 10 9 12 13 15 13 16 12 17 16 17 14 15 14 18 19 19 20 21 20 21 18 22 21 23 22 24 23 27 24 26 24 29 24 28 27 27 25 26 25 30 25 [出力例] 361 |
■参照サイト
HHKBプログラミングコンテスト2022 Winter(AtCoder Beginner Contest 282)