結果
問題 | No.2774 Wake up Record 2 |
ユーザー |
|
提出日時 | 2024-06-07 21:29:45 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 60 ms / 2,000 ms |
コード長 | 976 bytes |
コンパイル時間 | 2,371 ms |
コンパイル使用メモリ | 89,800 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-12-26 07:03:28 |
合計ジャッジ時間 | 2,404 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 |
ソースコード
#include <iostream>#include <vector>using namespace std;using ll = long long;constexpr int iINF = 1'000'000'000;constexpr ll llINF = 1'000'000'000'000'000'000;int main () {int N, K; cin >> N >> K;vector<int> A(N);for (int i = 0; i < N; i++) cin >> A[i];// 二分探索で寝る元気度を特定auto f = [&](ll x) {int count = 0;for (int i = 0; i < N; i++) if (A[i] <= x) count++;return count <= K;};ll ok = 1, ng = 1'000'000'000 + 10;while (1 < abs(ok - ng)) {ll mid = (ok + ng) / 2;if (f(mid)) {ok = mid;}else {ng = mid;}}vector<int> ans;for (int i = 1; i < N; i++) {if (A[i - 1] <= ok && ok < A[i]) {ans.push_back(i + 1);}}cout << ans.size() << "\n";for (int i = 0; i < ans.size(); i++) cout << ans[i] << (i == ans.size() - 1 ? "" : " ");cout << "\n";}