/ SeriousOJ /

Record Detail

Wrong Answer


  
# Status Time Cost Memory Cost
#1 Accepted 1ms 540.0 KiB
#2 Accepted 1ms 540.0 KiB
#3 Accepted 1ms 540.0 KiB
#4 Wrong Answer 2ms 540.0 KiB
#5 Wrong Answer 1ms 656.0 KiB

Code

#include<bits/stdc++.h>
using namespace std;
#define endl '\n'
#define ll long long int
#define vv vector<long long>
#define dq deque<long long>
#define mpp map<long long,long long>
#define ms multiset<long long>
#define ss set<long long>
#define vpp vector<pair<long long, long long>>
#define MOD 1000000007
#define YES cout<<"YES"<<endl
#define NO cout<<"NO"<<endl
#define all_v v.begin(),v.end()
#define all_a a.begin(),a.end()
#define sort_rev sort(v.rbegin(),v.rend())
#define all_n for(int i=0;i<n;i++)
#define print for(auto it:v)cout<<it<<" "
#define print_vp for(auto it:vp)cout<<it.first<<" "<<it.second<<endl
#define print_mp for(auto it:mp) cout<<it.first<<" "<<it.second<<endl
#define scan_v for(int i=0;i<n;i++)cin>>v[i]
#define scan_a for(int i=0;i<n;i++)cin>>a[i]
void requisite(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);cout.tie(nullptr);
}
int main() {
    requisite();
        ll n,k;
        cin>>n>>k;
        vv v(n);
        scan_v;
        ss s;
        vv a=v;
        sort(all_v);
        ll count=0;
        ll max_count=0;
    for (int i = 0; i < n; ++i) {
        if (a[i]!=v[i]){
            count= abs(i-(a[i]-1))+1;
            max_count= max(max_count,count);
            if (s.size()<k and max_count<=k){
                for (int j = i; j <i+count ; ++j) {
                    s.insert(a[j]);
                }
            }
        }
    }
//    cout<<max_count<<endl;
    if (max_count<=k){
        YES;
        for(auto it:s)cout<<it<<" ";
        cout<<endl;
    }
    else NO;
    return 0;
}

Information

Submit By
Type
Submission
Problem
P1120 Stairway to the Skyline
Contest
Brain Booster #7
Language
C++17 (G++ 13.2.0)
Submit At
2024-11-05 15:57:30
Judged At
2024-11-05 15:57:30
Judged By
Score
6
Total Time
2ms
Peak Memory
656.0 KiB