C++の練習を兼ねて, AtCoder Grand Contest 048 の 問題C (Penguin Skating) を解いてみた.
■感想.
1. 問題C は, 方針が見えなかったので, 解説を参考に実装したところ, AC版に到達出来た.
2. 個人的には, 解説のロジックで, 解けることが興味深く思った.
3. 区間を抽出する処理で, 尺取り法 の 復習ができたので 非常に良かったと思う.
4. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Grand Contest 048 解説 の 各リンク を ご覧下さい.
■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 |
// 解き直し. // https://atcoder.jp/contests/agc048/editorial/220 // C++20(GCC 12.2) #include <bits/stdc++.h> using namespace std; using LL = long long; using vi = vector<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--) int main(){ // 1. 入力情報. int N, L; scanf("%d %d", &N, &L); vi a(N), b(N); rep(i, N) scanf("%d", &a[i]); rep(i, N) scanf("%d", &b[i]); // 2. 空きマスの個数(以下, 解説通り). vi x(N + 1), y(N + 1); x[0] = a[0] - 1; y[0] = b[0] - 1; repx(i, 1, N){ x[i] = a[i] - a[i - 1] - 1; y[i] = b[i] - b[i - 1] - 1; } x[N] = L - a[N - 1]; y[N] = L - b[N - 1]; // 3. 区間[l, r]. bool ok = true; vi l(N + 1, 0), r(N + 1, 0); int xl = 0; rep(i, N + 1){ // y[i] が ゼロ. if(!y[i]) continue; // x[j] が ゼロとなるものを, 区間[l, r] から除外. repx(j, xl, N + 1){ if(!x[j]) ++xl; else break; } // y[i] が 正. int xr = 0, cur = 0; repx(j, xl, N + 1){ // 加算. cur += x[j]; // y[i] を 構成する x[l], ..., x[r] が 見つからなかった. if(cur > y[i]){ ok = false; break; } // y[i] を 構成する x[l], ..., x[r] が 見つかった. if(cur == y[i]){ xr = j; l[i] = xl; r[i] = xr; break; } } // 条件を満たさない場合. if(!ok) break; // 次へ. xl = xr + 1; } // rep(i, N + 1) printf("%d %d\n", l[i], r[i]); // 4. 達成不可能な場合. if(!ok){ puts("-1"); return 0; } // 5. 最小回数. LL ans = 0; rep(i, N + 1) if(y[i]) ans += max(i - l[i], 0) + max(r[i] - i, 0); // 6. 出力. 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 |
[入力例] 4 11 3 4 6 10 1 5 6 11 [出力例] 3 ※AtCoderのテストケースより [入力例] 1 3 1 2 [出力例] -1 ※AtCoderのテストケースより [入力例] 10 1000000000 65110170 68805223 123016442 275946481 661490312 760727752 764540566 929355340 930658577 947099792 1 2 123016442 661490311 929355337 930658574 999999997 999999998 999999999 1000000000 [出力例] 13 ※AtCoderのテストケースより [入力例] 6 25 4 7 12 15 18 23 1 5 8 13 16 25 [出力例] 6 [入力例] 10 60 5 12 18 19 27 28 35 39 47 55 1 2 7 19 20 21 22 30 54 60 [出力例] 11 [入力例] 20 80 1 2 3 4 5 8 10 17 22 23 30 32 33 40 48 52 55 60 69 75 1 4 5 6 14 15 16 21 28 29 37 38 39 50 51 54 72 73 79 80 [出力例] 21 |
■参照サイト
AtCoder Grand Contest 048