Submission #1368699

#TimeUsernameProblemLanguageResultExecution timeMemory
1368699po_rag526Table Tennis (info1cup20_tabletennis)C++20
49 / 100
3095 ms2060 KiB
#include <bits/stdc++.h>
using namespace std;
#define int long long
#define fi first
#define se second
#define pb push_back

int n,k, ar[150001];

vector<int> val(int l, int r){
    int rem=l-1 + n+k-r;
    if(rem>k) return vector<int>(1,-1);

    vector<int> idx;
    int sum=ar[l]+ar[r];
    while(l<r){
        if(ar[l]+ar[r]==sum){l++; r--;}
        else if(ar[l]+ar[r]>sum) idx.pb(r--);
        else if(ar[l]+ar[r]<sum) idx.pb(l++);

        if(r-l-1==k-rem-(int)idx.size()){
            for(int i=l+1;i<=r-1;i++) idx.pb(i);
            return idx;
        }

        if(rem+(int)idx.size()>k) return vector<int>(1,-1);
    }

    if(l==r) idx.pb(l);

    return (rem+(int)idx.size()==k ? idx : vector<int>(1,-1));
}

signed main(){
    ios_base::sync_with_stdio(false); cin.tie(nullptr);

    cin >> n >> k;
    for(int i=1;i<=n+k;i++) cin >> ar[i];

    for(int l=1;l<=k+1;l++){
        for(int r=l+n-1;r<=n+k;r++){
            vector<int> res=val(l,r);

            if(!(!res.empty() && res[0]==-1)){
                int ite=0;
                sort(res.begin(), res.end());
                for(int i=l;i<=r;i++){
                    if(ite<res.size() && res[ite]==i){ite++; continue;}
                    cout << ar[i] << ' ';
                }

                // for(int el : res) cout << el << ' ';
                // cout << '\n';
                // cout << l << ' ' << r;
                return 0;
            }
        }
    }
}
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...
#Result Execution timeMemoryGrader output
Fetching results...