C++の練習を兼ねて, AtCoder Regular Contest 041 の 問題C (ウサギ跳び) を解いてみた.
■感想.
1. 問題Cは, 方針が見えなかったので, 解説を参考に, AC版に到達できたと思う.
2. 実装に苦労したものの, 個人的には, 解説のロジックが, 不思議な印象を受けた.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Regular Contest 041 解説 を ご覧下さい.
■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 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 |
// 解き直し. // https://www.slideshare.net/chokudai/arc041 // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; using LL = long long; using P = pair<int, int>; using vp = vector<P>; using vvp = vector<vp>; #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 pof pop_front #define pob pop_back #define puf push_front #define pub push_back #define a first #define b second int main(){ // 1. 入力情報. int N; LL L; scanf("%d %lld", &N, &L); vp v; rep(i, N){ int x; char c[11]; scanf("%d %s", &x, c); v.pub({x, ((c[0] == 'R') ? 1 : -1)}); } // 2. ウサギを区間に分割. vvp rabbits; deque<P> dq; for(auto &p : v){ // 空. if(dq.empty()){ dq.pub(p); continue; } // 背中合わせでない. P r = dq.back(); if(!(r.b == -1 && p.b == 1)) dq.pub(p); // 背中合わせである. if(r.b == -1 && p.b == 1){ vp rabbit; while(!dq.empty()){ rabbit.pub(dq.front()); dq.pof(); } rabbits.pub(rabbit); dq.pub(p); } } // 3. 残分を追加. vp rabbit; while(!dq.empty()){ rabbit.pub(dq.front()); dq.pof(); } rabbits.pub(rabbit); /* for(auto &p : rabbits){ for(auto &q : p) printf("%d %d\n", q.a, q.b); puts("------"); } */ // 4. 区間ごとに, 集計. LL ans = 0; for(auto &p : rabbits){ // 4-1. 左向き, 右向きの個数. int l = 0, r = 0; LL lMin = L + 1, rMax = 0, t = 0; for(auto &q : p){ if(q.b == -1){ ++l; lMin = min(lMin, (LL)q.a); } if(q.b == +1){ ++r; rMax = max(rMax, (LL)q.a); } } // 4-2. 左向き >= 右向き. if(l >= r){ rep(i, r) t += (rMax - i - p[r - 1 - i].a); rep(i, l) t += (p[r + i].a - (i + 1) - rMax); ans += t; } // 4-3. 左向き < 右向き. if(l < r){ rep(i, r) t += (lMin - (i + 1) - p[r - 1 - i].a); rep(i, l) t += (p[r + i].a - i - lMin); ans += t; } } // 5. 出力. 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 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 |
[入力例] 1 5 1 R [出力例] 4 ※AtCoderのテストケースより [入力例] 4 5 1 R 3 L 4 L 5 L [出力例] 3 ※AtCoderのテストケースより [入力例] 4 10 1 L 5 R 6 L 10 R [出力例] 0 ※AtCoderのテストケースより [入力例] 1 7 7 L [出力例] 6 [入力例] 2 5 2 R 4 L [出力例] 1 [入力例] 5 20 5 R 8 R 12 L 15 L 17 L [出力例] 16 [入力例] 7 25 2 R 5 L 8 R 11 R 12 L 15 R 17 L [出力例] 5 [入力例] 19 55 2 R 5 L 8 R 11 R 12 L 15 R 17 L 19 R 22 R 27 R 29 R 33 R 35 L 40 L 43 L 48 R 49 L 52 L 54 L [出力例] 50 |
■参照サイト
AtCoder Regular Contest 041