C++の練習を兼ねて, AtCoder Beginner Contest 222 の 問題E (Red and Blue Tree) を解いてみた.
■感想.
1. 問題Eは, 方針が見えなかったので, 解説を参考に提出して, ようやく, AC版に到達出来た.
2. 幅優先探索の復習が出来たので, 非常に良かったと思う.
3. 苦手な動的計画法の訓練を積めたので, 非常に良かったと思う.
4. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 222 解説 の 各リンク を ご覧下さい.
■C++版プログラム(問題E/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 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 150 |
// 解き直し. // https://atcoder.jp/contests/abc222/editorial/2751 // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using LL = long long; using vi = vector<int>; using vvi = vector<vi>; using P = pair<int, int>; #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 const LL MOD = 998244353; int a[101], c[1010]; LL dp[1010][101010]; int main(){ // 1. 入力情報. int N, M, K; scanf("%d %d %d", &N, &M, &K); rep(i, M){ scanf("%d", &a[i]); a[i]--; } vvi G(N); map<P, int> E; rep(i, N - 1){ int u, v; scanf("%d %d", &u, &v); u--; v--; G[u].pb(v); G[v].pb(u); // 辺 i は, (u[i], v[i]) or (v[i], u[i]) と 読み替え. E[{u, v}] = i; E[{v, u}] = i; } // 2. bfs. // https://ja.wikipedia.org/wiki/幅優先探索 auto bfs = [&](vvi &G, int s, int* d){ // 空のキュー. queue<int> q; // 探索地点 s をキュー q に追加. q.push(s); while(!q.empty()){ // キューから取り出す. int u = q.front(); q.pop(); // 隣接頂点をチェック. for(auto &e : G[u]){ // 未訪問の頂点を処理. if(d[e]) continue; if(e != s) d[e] = d[u] + 1, q.push(e); } } return; }; // 3. 最短経路. auto f = [&](vvi &G, int g, int* d){ // 空のキュー. queue<int> q; // 探索地点 g をキュー q に追加. q.push(g); while(!q.empty()){ // キューから取り出す. int u = q.front(); q.pop(); // 隣接頂点をチェック. for(auto &e : G[u]){ if(d[e] == d[u] - 1){ // 距離が 1 小さい頂点を追加. q.push(e); // 最短経路を構成する辺とみて, カウントアップ. c[E[{e, u}]]++; } } } return; }; // 4. 各辺を通過する数を計算. rep(i, M - 1){ // 辺情報. int s = a[i]; int g = a[i + 1]; // 最短距離. int d[1010]; rep(j, 1010) d[j] = 0; bfs(G, s, d); // puts("\n----- bfs start -----"); // rep(j, N) printf("%d ", d[j]); // puts("\n----- bfs end -----"); // 最短経路. f(G, g, d); // puts("----- shortest path start -----"); // rep(j, N - 1) printf("%d ", c[j]); // puts("\n----- shortest path end -----"); } // 5. 例外チェック. // -> (K + S) / 2 が 整数でない, 0未満. int S = 0; rep(i, N - 1) S += c[i]; int x = (K + S) / 2; // printf("S=%d K=%d x=%d\n", S, K, x); if(x * 2 != K + S || x < 0){ puts("0"); return 0; } // 6. dp更新. rep(i, 1010) dp[i][0] = 1; rep(i, N - 1){ // 前回分反映. if(i) rep(j, 101010) dp[i][j] = dp[i - 1][j]; // 今回分反映. repr(j, 101009, 0){ int at = j + c[i]; if(at <= 101009){ if(dp[i][j]){ dp[i][at] += dp[i][j]; dp[i][at] %= MOD; } } } } // rep(i, N - 1){ // rep(j, 10) printf("%lld ", dp[i][j]); // puts(""); // } // 7. 出力. printf("%lld\n", dp[N - 2][x]); 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 150 151 152 |
[入力例] 4 5 0 2 3 2 1 4 1 2 2 3 3 4 [出力例] 2 ※AtCoderテストケースより [入力例] 3 10 10000 1 2 1 2 1 2 2 1 1 2 1 2 1 3 [出力例] 0 ※AtCoderテストケースより [入力例] 10 2 -1 1 10 1 2 2 3 3 4 4 5 5 6 6 7 7 8 8 9 9 10 [出力例] 126 ※AtCoderテストケースより [入力例] 5 8 -1 1 4 1 4 2 1 3 5 1 2 4 1 3 1 1 5 [出力例] 2 ※AtCoderテストケースより [入力例] 2 2 1 2 1 1 2 [出力例] 1 [入力例] 7 7 3 2 1 6 4 5 7 3 2 1 2 3 1 7 1 5 3 6 4 3 [出力例] 8 [入力例] 13 36 2 2 10 11 3 6 7 13 4 12 7 5 12 8 6 11 10 9 11 5 13 4 3 12 6 6 2 1 11 10 4 4 13 7 3 7 2 2 1 2 7 2 3 2 8 8 9 10 8 10 11 10 6 3 4 5 3 5 13 5 12 [出力例] 226 [入力例] 21 77 6 18 9 4 6 8 18 9 9 15 7 4 19 15 19 7 20 9 7 1 16 14 6 13 10 14 16 5 1 7 7 15 18 18 10 15 5 2 19 1 1 9 7 9 16 18 5 8 17 18 12 1 3 18 20 10 9 9 18 16 14 8 15 12 15 6 11 5 2 2 12 14 1 19 12 21 13 10 7 1 4 1 6 1 2 3 2 5 2 7 7 8 11 10 12 10 9 10 9 8 9 16 16 17 18 17 8 13 13 15 13 14 8 19 19 20 19 21 [出力例] 12318 [入力例] 30 100 -8 4 8 16 20 19 2 28 29 12 14 22 13 9 29 24 17 24 1 17 19 19 2 20 10 2 25 9 21 5 4 11 27 20 6 28 24 6 29 27 22 16 8 23 5 21 27 5 1 29 26 30 11 10 20 4 20 28 22 19 19 8 6 14 1 5 5 14 25 4 29 3 29 19 21 24 30 8 24 7 28 3 21 18 4 6 5 17 2 15 29 12 5 29 13 18 7 22 3 23 2 1 5 2 1 15 1 14 1 4 1 3 1 3 13 3 12 12 22 12 23 16 15 17 15 16 27 16 26 26 29 26 28 2 30 2 6 2 7 7 24 25 7 5 8 9 5 18 14 14 19 4 10 4 11 20 11 21 11 [出力例] 5324136 |
■参照サイト
エクサウィザーズプログラミングコンテスト2021(AtCoder Beginner Contest 222)