C - Make It Good

cpcodeforcesdiv3greedybinary search

最終更新日

Table of Contents

Loading...

問題

https://codeforces.com/contest/1385/problem/C (新しいタブで開く)

問題文

長さ nn の数列 aa の先頭からいくつかの要素を削除して、 いい配列 にしたい。
削除する必要がある要素の最小値は?

数列 bbいい配列 であるとは、bb が空になるまで以下の操作を繰り返して、最初は空の配列 cc を広義単調増加にできることをいう。

bb の先頭か末尾のどちらかから要素を選ぶ。
選んだ要素を bb から削除し、 cc の末尾に加える。

制約

サンプル

5 4 1 2 3 4 7 4 3 3 8 4 5 2 3 1 1 1 7 1 3 1 4 5 3 2 5 5 4 3 2 3
0 4 0 2 3

考察

決め打ち二分探索 + 貪欲法した。解説は全然そんなことしてなかった

具体的には、 kk を決め打ちして、 aa の先頭 kk 個を削除してから、貪欲に いい配列 にできるか判定する。
先頭/末尾の要素のうち、小さいほうから順番に cc に加えていくのが最適。

コード

https://codeforces.com/contest/1385/submission/126231155 (新しいタブで開く)

bool chk(deque<int> dq, int start) { int prev = 0; if (start >= dq.size()) return false; while (start--) dq.pop_front(); while (!dq.empty()) { int fr = dq.front(); int ba = dq.back(); if (fr >= prev && fr <= ba) { prev = fr; dq.pop_front(); } else if (ba >= prev && ba <= fr) { prev = ba; dq.pop_back(); } else if (fr >= prev) { prev = fr; dq.pop_front(); } else if (ba >= prev) { prev = ba; dq.pop_back(); } else { break; } } return dq.empty(); } void solve() { int n; cin >> n; deque<int> dq; rep(i, n) { int a; cin >> a; dq.push_back(a); } int ok = n - 1, ng = -1; while (abs(ok - ng) > 1) { int mid = (ok + ng) / 2; if (chk(dq, mid)) { ok = mid; } else { ng = mid; } } cout << ok << endl; }