結果
問題 | No.2774 Wake up Record 2 |
ユーザー |
![]() |
提出日時 | 2024-06-07 21:39:51 |
言語 | C++17(gcc12) (gcc 12.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 29 ms / 2,000 ms |
コード長 | 1,252 bytes |
コンパイル時間 | 5,369 ms |
コンパイル使用メモリ | 267,432 KB |
実行使用メモリ | 7,288 KB |
最終ジャッジ日時 | 2024-12-26 07:25:46 |
合計ジャッジ時間 | 5,219 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 |
ソースコード
#include <bits/stdc++.h>#include <atcoder/all>#define rep(i,n) for(int (i)=0;(i)<(n);(i)++)#define rep2(i,m,n) for(int (i)=(m);(i)<(n);(i)++)#define rep2ll(i,m,n) for(ll (i)=(m);(i)<(n);(i)++)#define ALL(obj) (obj).begin(), (obj).end()#define rALL(obj) (obj).rbegin(), (obj).rend()using namespace std;using ll = long long;using P = pair<ll,ll>;using mint = atcoder::modint998244353;using VL = vector<ll>;using VVL = vector<VL>;using VVVL = vector<VVL>;using VM = vector<mint>;using VVM = vector<VM>;using VVVM = vector<VVM>;using VD = vector<double>;using VVD = vector<VD>;using VVVD = vector<VVD>;template<class T>bool chmax(T &a, const T &b) { if (a<b) { a=b; return 1; } return 0; }template<class T>bool chmin(T &a, const T &b) { if (b<a) { a=b; return 1; } return 0; }int main(){ios::sync_with_stdio(false);cin.tie(0);ll n, k; cin>>n>>k;VL a(n); rep(i,n) cin>>a[i];vector<P> ai;rep(i,n) ai.emplace_back(a[i], i);sort(ALL(ai));VL b(n, 0);rep(i,k) b[ai[i].second] = 1;VL ans;rep(i,n-1){if(b[i] == 1 && b[i+1] == 0){ans.push_back(i+2);}}cout<<ans.size()<<'\n';for(auto x:ans) cout<<x<<' ';cout<<'\n';return 0;}