Submission #1173997

#TimeUsernameProblemLanguageResultExecution timeMemory
1173997ezzzayPilot (NOI19_pilot)C++20
89 / 100
1100 ms95832 KiB
#include<bits/stdc++.h>
using namespace std;
#define ff first
#define ss second
#define pb push_back
#define int long long
const int N=1e6+5;
int a[N];
int ans[N];
signed main(){
    int n,q;
    cin>>n>>q;
    int H= (n*(n-1))/2+n;
    vector<pair<int,int>>vc;
    for(int i=1;i<=n;i++){
        cin>>a[i];
        vc.pb({a[i],i});
    }
    vector<pair<int,int>>qr(q);
    for(int i=0;i<q;i++){
        cin>>qr[i].ff;
        qr[i].ss=i;
    }
    sort(qr.begin(),qr.end() ,greater< pair<int,int>>());
    sort(vc.begin(),vc.end());
    set<int>st={0,n+1};
    for(auto p:qr){
        int x=p.ff,id=p.ss;
        
        while(vc.size() and vc.back().ff>x){
            
            int idx=vc.back().ss;
            int R=*st.upper_bound(idx);
            int L= *prev(st.lower_bound(idx));
            int p=R-L-1;
            
            H-=(p*(p-1)/2+p);
            p=idx-L-1;
            H+=(p*(p-1)/2+p);
            p=R-idx-1;
            H+=(p*(p-1)/2+p);
            
            st.insert(idx);
            vc.pop_back();
        }
        ans[id]=H;
    }
    
    for(int i=0;i<q;i++){
        cout<<ans[i]<<'\n';
    }
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...