結果
問題 | No.2774 Wake up Record 2 |
ユーザー | NAMIDAIRO |
提出日時 | 2024-06-07 21:31:22 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 48 ms / 2,000 ms |
コード長 | 1,134 bytes |
コンパイル時間 | 1,381 ms |
コンパイル使用メモリ | 123,800 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-12-26 07:08:13 |
合計ジャッジ時間 | 2,479 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 |
ソースコード
#include <iostream> #include <vector> #include <algorithm> #include <map> #include <queue> #include <set> #include <random> #include <iomanip> #include <string> #include <cmath> #include <complex> using namespace std; typedef long long ll; #define rep(i, n) for(int i = 0; i < (n); i++) template<class T> using vi = vector<T>; template<class T> using vii = vector<vi<T>>; template<class T> using viii = vector<vii<T>>; template<class T> using viiii = vector<viii<T>>; using P = pair<ll, int>; void chmin(ll & x, ll y) { x = min(x, y); } void chmax(ll& x, ll y) { x = max(x, y); } int main() { int n, k; cin >> n >> k; vi<int> a(n); rep(i, n) cin >> a[i]; vi<int> b = a; sort(b.begin(), b.end()); int thre = b[k - 1]; string s; rep(i, n) { if (a[i] <= thre) s.push_back('x'); else s.push_back('o'); } vi<int> ans; rep(i, n - 1) { if (s[i] == 'x' && s[i + 1] == 'o') ans.push_back(i + 2); } int sz = (int)ans.size(); cout << sz << endl; rep(i, sz) cout << ans[i] << " \n"[i == sz - 1]; if (sz == 0) cout << endl; return 0; }