C++の練習を兼ねて, AtCoder Regular Contest 125 の 問題C (LIS to Original Sequence) を解いてみた.
■感想.
1. 問題Cは, 方針が見えなかったので, 解説を参考に, AC版に到達できたと思う.
2. 個人的には, 解説のロジックで, 辞書順最小 の 順列 を 計算できることが, 非常に興味深く思った.
3. 引き続き, 時間を見つけて, 過去問の学習を進めていきたいと思う.
本家のサイト AtCoder Regular Contest 125 解説 の 各リンク を ご覧下さい.
■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 |
// 解き直し. // https://atcoder.jp/contests/arc125/editorial/2499 // 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 a[202020]; int main(){ // 1. 入力情報. int N, K; scanf("%d %d", &N, &K); rep(i, N) scanf("%d", &a[i]); // 2. 未使用リスト. set<int> st; rep(i, N) st.insert(i + 1); // 3. 順列を構築. vi ans; rep(i, N){ // K が 1. if(--K == 0){ for(auto i = st.rbegin(), e = st.rend(); i != e; ++i) ans.pb(*i); break; } // 整数列 A の 要素. if(st.count(a[i])){ // 保存. ans.pb(a[i]); // 例外. int f = *st.begin(); if(a[i] > f){ ans.pb(f); st.erase(f); } } // 未使用リストから除外. st.erase(a[i]); } // 4. 出力 rep(i, N) printf("%d%s", ans[i], (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 |
[入力例] 3 2 2 3 [出力例] 2 1 3 ※AtCoderのテストケースより [入力例] 5 1 4 [出力例] 5 4 3 2 1 ※AtCoderのテストケースより [入力例] 3 1 3 [出力例] 3 2 1 [入力例] 7 4 3 5 6 7 [出力例] 3 1 5 2 6 4 7 [入力例] 10 5 2 4 6 8 9 [出力例] 2 1 4 3 6 5 8 7 10 9 [入力例] 20 7 3 4 7 8 9 12 15 [出力例] 3 1 4 2 7 5 8 6 9 12 10 20 19 18 17 16 15 14 13 11 [入力例] 50 10 5 7 11 25 28 31 34 43 47 48 [出力例] 5 1 7 2 11 3 25 4 28 6 31 8 34 9 43 10 47 12 50 49 48 46 45 44 42 41 40 39 38 37 36 35 33 32 30 29 27 26 24 23 22 21 20 19 18 17 16 15 14 13 |
■参照サイト
AtCoder Regular Contest 125