C++の練習を兼ねて, AtCoder Beginner Contest 221 の 問題G (Jumping sequence) を解いてみた.
■感想.
1. 問題Gは, 方針が見えなかったので, 解説を参考にして, ようやく, AC版に到達出来た.
2. 苦手なdpの訓練を積めたこと, および, bitsetの使い方を訓練できたのが非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Beginner Contest 221 解説 の 各リンク を ご覧下さい.
■C++版プログラム(問題G/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 |
// C++(GCC 9.2.1) // 解き直し. // https://atcoder.jp/contests/abc221/editorial/2724 #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 #define all(x) x.begin(), x.end() int D[2020]; // bitset<55> dp[11]; bitset<3636363> dp[2020]; int main(){ // 1. 入力情報. int N, A, B, S = 0; scanf("%d %d %d", &N, &A, &B); rep(i, N){ scanf("%d", &D[i]); S += D[i]; } // 2. 例外をチェック. int D1 = (S + A + B) / 2; int D2 = (S + A - B) / 2; bool ok = true; ok &= ((S + A + B) == D1 * 2); ok &= ((S + A - B) == D2 * 2); ok &= (D1 >= 0); ok &= (D1 <= S); ok &= (D2 >= 0); ok &= (D2 <= S); if(!ok){ puts("No"); return 0; } // 3. dp更新. dp[0][0] = true; rep(i, N){ // 3-1. 今回分設定. dp[i + 1] = dp[i]; // 3-2. dpを左シフトしたものを考慮. dp[i + 1] <<= D[i]; // 3-3. 前回分を反映. dp[i + 1] |= dp[i]; } // 4. 例外をチェック. if(!dp[N][D1] || !dp[N][D2]){ puts("No"); return 0; } // 5. 経路の復元. // -> 以下のような変換ルールが, 発生したと見做す. // 上: (0, +D) -> (+D, -D) -- +D --> (+2D, 0) -- /2 --> (+D, 0) // 下: (0, -D) -> (-D, +D) -- +D --> (0, +2D) -- /2 --> (0, +D) // 左: (-D, 0) -> (-D, -D) -- +D --> (0, 0) -- /2 --> (0, 0) // 右: (+D, 0) -> (+D, +D) -- +D --> (+2D, +2D) -- /2 --> (+D, +D) string ans = ""; int curX = D1, curY = D2; repr(i, N - 1, 0){ // 上. if(curX - D[i] >= 0 && dp[i][curX - D[i]] && dp[i][curY]){ // 経路情報. ans.pb('U'); // 今回分反映. curX -= D[i]; curY -= 0; // 次へ. continue; } // 下. if(curY - D[i] >= 0 && dp[i][curX] && dp[i][curY - D[i]]){ // 経路情報. ans.pb('D'); // 今回分反映. curX -= 0; curY -= D[i]; // 次へ. continue; } // 左. if(dp[i][curX] && dp[i][curY]){ // 経路情報. ans.pb('L'); // 今回分反映. curX -= 0; curY -= 0; // 次へ. continue; } // 右. if(curX - D[i] >= 0 && curY - D[i] >= 0 && dp[i][curX - D[i]] && dp[i][curY - D[i]]){ // 経路情報. ans.pb('R'); // 今回分反映. curX -= D[i]; curY -= D[i]; // 次へ. continue; } } // 6. 反転. reverse(all(ans)); // 7. 出力. puts("Yes"); 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 38 39 40 41 42 43 44 45 46 47 48 49 |
[入力例] 3 2 -2 1 2 3 [出力例] Yes LDR ※AtCoderのテストケースより ※但し, 上記のプログラムでは, 以下の内容が出力される. Yes URD [入力例] 2 1 0 1 6 [出力例] No ※AtCoderのテストケースより [入力例] 5 6 7 1 3 5 7 9 [出力例] Yes LRLUR ※AtCoderのテストケースより ※但し, 上記のプログラムでは, 以下の内容が出力される. Yes LUDRU [入力例] 11 15 13 3 1 4 1 5 9 2 6 5 3 5 [出力例] Yes DRRRDRUUUUU [入力例] 31 62 27 7 6 60 1 6 8 3 7 9 3 3 19 9 8 8 9 3 54 4 4 3 27 18 5 3 3 7 19 5 55 13 [出力例] Yes RRDDRRDRRDDRDDUDUDUUUUUUUUUUUUU |
■参照サイト
AtCoder Beginner Contest 221