C++の練習を兼ねて, AtCoder Beginner Contest 178 の 問題F (Contrast) を解いてみた.
■感想.
1. 方針が見えなかったので, 解説を参照して実装して, ようやく, AC版となった.
2. 時間を見つけて, 引き続き, 過去問を振り返っていきたいと思う.
本家のサイト
AtCoder Beginner Contest 178 (F – Contrast)解説
をご覧下さい.
■C++版プログラム(問題F／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 |
// 解き直し. // https://atcoder.jp/contests/abc178/editorial/84 // C++(GCC 9.2.1) #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--) const int MAX = 202020; int A[MAX], B[MAX], a[MAX], C[MAX], b[MAX], D[MAX]; int main() { // 1. 入力情報取得. int N; scanf("%d", &N); rep(i, N){ scanf("%d", &A[i]); a[A[i] - 1]++; } rep(i, N){ scanf("%d", &B[i]); b[B[i] - 1]++; } // 2. 条件を満たさないパターンを確認. rep(i, N){ if(a[i] + b[i] > N){ puts("No"); return 0; } } // 3. A, B に出現する i以下 の 値 の個数. rep(i, N) C[i + 1] = C[i] + a[i]; rep(i, N) D[i + 1] = D[i] + b[i]; // 4. C[i] - D[i - 1] の 最大値. int x = -1; repx(i, 1, N + 1) x = max(x, C[i] - D[i - 1]); assert(x != -1); // 5. 出力. puts("Yes"); rep(i, N){ printf("%d", B[(i - x + N) % N]); printf("%s", (i < N - 1) ? " " : "\n"); } 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 |
[入力例] 6 1 1 1 2 2 3 1 1 1 2 2 3 [出力例] Yes 2 2 3 1 1 1 ※AtCoderテストケースより [入力例] 3 1 1 2 1 1 3 [出力例] No ※AtCoderテストケースより [入力例] 4 1 1 2 3 1 2 3 3 [出力例] Yes 3 3 1 2 ※AtCoderテストケースより [入力例] 2 1 2 1 2 [出力例] Yes 2 1 [入力例] 3 1 2 3 1 2 2 [出力例] Yes 2 1 2 [入力例] 5 1 2 2 2 3 1 2 2 3 3 [出力例] Yes 2 3 3 1 2 [入力例] 10 2 4 6 6 7 7 7 8 9 10 1 1 1 2 5 6 7 7 8 10 [出力例] Yes 10 1 1 1 2 5 6 7 7 8 [入力例] 17 1 1 1 2 2 3 3 3 3 3 6 6 7 8 11 11 12 3 3 3 3 3 3 6 6 6 7 7 8 8 8 9 10 11 [出力例] Yes 6 6 7 7 8 8 8 9 10 11 3 3 3 3 3 3 6 [入力例] 23 2 2 3 3 4 4 4 5 5 5 5 5 6 6 7 7 7 7 7 8 8 10 10 1 1 1 1 1 2 2 2 3 3 3 3 4 4 6 6 6 7 7 8 10 10 10 [出力例] Yes 10 10 10 1 1 1 1 1 2 2 2 3 3 3 3 4 4 6 6 6 7 7 8 |
■参照サイト
AtCoder Beginner Contest 178