C++の練習を兼ねて, AtCoder Regular Contest 140 の 問題C (ABS Permutation (LIS ver.)) を解いてみた.
■感想.
1. 問題Cは, 方針が見えなかったので, 解説を参考にして, ようやく, AC版に到達出来た.
2. 解説のロジックで, 条件を満たす順列を抽出できることが, 非常に印象的だった.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Regular Contest 140 解説 の 各リンク を ご覧下さい.
■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 |
// 解き直し. // https://atcoder.jp/contests/arc140/editorial/3893 // C++(GCC 9.2.1) #include <bits/stdc++.h> using namespace std; 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--) #define pb push_back int main(){ // 1. 入力情報. int N, X; scanf("%d %d", &N, &X); int M = N / 2; // 2. N が 奇数. if(N & 1){ // 2-1. X が M + 1. if(X == M + 1){ int cur = X, sign = -1; rep(i, N){ sign *= -1; cur += i * sign; printf("%d%s", cur, (i < N - 1) ? " " : "\n"); } } // 2-2. X が M + 1 以外. if(X != M + 1){ vi v; rep(i, N) if(X != i + 1) v.pb(i + 1); int cur = M - 1, sign = -1; printf("%d ", X); rep(i, N - 1){ cur += i * sign; printf("%d%s", v[cur], (i < N - 2) ? " " : "\n"); sign *= -1; } } } // 3. N が 偶数. if(!(N & 1)){ // 3-1. X が M or M + 1. if(X == M || X == M + 1){ int cur = X, sign = (X == M) ? -1 : 1; rep(i, N){ cur += i * sign; sign *= -1; printf("%d%s", cur, (i < N - 1) ? " " : "\n"); } } // 3-2. X が M, M + 1 以外. if(X != M && X != M + 1){ vi v; rep(i, N) if(X != i + 1) v.pb(i + 1); int cur = M - 1, sign = -1; printf("%d %d ", X, v[cur]); rep(i, N - 2){ cur += (i + 1) * sign; printf("%d%s", v[cur], (i < N - 3) ? " " : "\n"); sign *= -1; } } } 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 |
[入力例] 3 2 [出力例] 2 1 3 ※AtCoderのテストケースより [入力例] 3 1 [出力例] 1 2 3 ※AtCoderのテストケースより [入力例] 6 1 [出力例] 1 4 3 5 2 6 [入力例] 6 3 [出力例] 3 4 2 5 1 6 [入力例] 6 4 [出力例] 4 3 5 2 6 1 [入力例] 7 3 [出力例] 3 4 5 2 6 1 7 [入力例] 7 4 [出力例] 4 3 5 2 6 1 7 [入力例] 7 5 [出力例] 5 3 4 2 6 1 7 [入力例] 30 15 [出力例] 15 16 14 17 13 18 12 19 11 20 10 21 9 22 8 23 7 24 6 25 5 26 4 27 3 28 2 29 1 30 [入力例] 123 62 [出力例] 62 61 63 60 64 59 65 58 66 57 67 56 68 55 69 54 70 53 71 52 72 51 73 50 74 49 75 48 76 47 77 46 78 45 79 44 80 43 81 42 82 41 83 40 84 39 85 38 86 37 87 36 88 35 89 34 90 33 91 32 92 31 93 30 94 29 95 28 96 27 97 26 98 25 99 24 100 23 101 22 102 21 103 20 104 19 105 18 106 17 107 16 108 15 109 14 110 13 111 12 112 11 113 10 114 9 115 8 116 7 117 6 118 5 119 4 120 3 121 2 122 1 123 |
■参照サイト
AtCoder Regular Contest 140