C++の練習を兼ねて, エクサウィザーズ 2019 の 問題C (Snuke the Wizard) を解いてみた.
■感想.
1. 問題Cは, 方針が見えなかったので, 解説を参考に, AC版に到達できたと思う.
2. 二分探索の復習が出来たので, 非常に良かったと思う.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト エクサウィザーズ 2019 解説 を ご覧下さい.
■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 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 |
// 解き直し. // https://img.atcoder.jp/exawizards2019/editorial.pdf // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using P = pair<char, char>; using vp = vector<P>; #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, Q; char s[202020]; scanf("%d %d %s", &N, &Q, s); string S(s); S = '#' + S + '#'; vp op; rep(i, Q){ char t[11], d[11]; scanf("%s %s", t, d); op.pb({t[0], d[0]}); } // 2. 評価関数. // -> マス 0 に到達するか否か. auto f = [&](int m) -> bool { // 2-1. 呪文を, Q回詠唱. char c = S[m]; int p = m, pMin = N + 1; for(auto &o : op){ if(o.a != c) continue; p += (o.b == 'R') ? +1 : -1; if(p >= 1 && p <= N) c = S[p]; pMin = min(pMin, p); } // 2-2. 判定結果. return (pMin > 0); }; // 3. 二分探索で確認してみる. int l = 0, h = N + 1, m, ans = 0; while(l + 1 < h){ int m = (h + l) >> 1; if(f(m)) h = m; else l = m; // printf("h=%d l=%d m=%d\n", h, l, m); } ans = l; // 4. 評価関数. // -> マス (N + 1) に到達するか否か. auto g = [&](int m) -> bool { // 4-1. 呪文を, Q回詠唱. char c = S[m]; int p = m, pMax = 0; for(auto &o : op){ if(o.a != c) continue; p += (o.b == 'R') ? +1 : -1; if(p >= 1 && p <= N) c = S[p]; pMax = max(pMax, p); } // 4-2. 判定結果. return (pMax < N + 1); }; // 5. 二分探索で確認してみる. l = 0, h = N + 1, m; while(l + 1 < h){ int m = (h + l) >> 1; if(g(m)) l = m; else h = m; // printf("h=%d l=%d m=%d\n", h, l, m); } ans = max(0, l - ans); // 6. 出力. 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 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 |
[入力例] 3 4 ABC A L B L B R A R [出力例] 2 ※AtCoderテストケースより [入力例] 8 3 AABCBDBA A L B R A R [出力例] 5 ※AtCoderテストケースより [入力例] 10 15 SNCZWRCEWB B R R R E R W R Z L S R Q L W L B R C L A L N L E R Z L S L [出力例] 3 ※AtCoderテストケースより [入力例] 5 5 ABABA A R B R A R B R A L [出力例] 2 [入力例] 11 15 ABBCCDCCBBA D R C R C R A L A R B L C R B R B R A R C L B L B L A L D R [出力例] 2 [入力例] 50 50 IAHJAFEIHHBDIAJMECDLEMGBJMMDKILFLJFLBKMIEEDDDMMDFA E L L L D L C L E L M L N R A R J L E R A L E R I L D L B L H R H L M R H L D R I L F R E L F L E L A L B L J L A R H L A L I L D R M R D R G R D R F R A R F L A L J R H L J L D R A L I L L L E R K L [出力例] 37 |
■参照サイト
エクサウィザーズ 2019