Submission #885736

#	Time	Username	Problem	Language	Result	Execution time	Memory
885736		Ninedesu	Pilot (NOI19_pilot)	C++14	78 / 100	1059 ms	3616 KiB

pilot

#include<bits/stdc++.h>
#define ll long long
#define pii pair<int,int>
using namespace std;

const int N=1e6+1;
int n,q;
int arr[N];
bool chk=true;
pii qst[N];

int main(){
    ios_base::sync_with_stdio(0),cin.tie(0);
    cin >> n >> q;
    for(int i=1; i<=n; i++){
        cin >> arr[i];
        if(arr[i]<arr[i-1])chk=false;
    }
    if(chk==true){
        while(q--){
            int h;
            cin >> h;
            int idx=upper_bound(arr+1,arr+n+1,h)-arr-1;
            cout << ((ll)idx*(idx+1))/2 << '\n';
        }
    }
    else{
        while(q--){
            int h;
            ll ans=0,cnt=0;
            cin >> h;
            for(int i=1; i<=n; i++){
                if(arr[i]<h+1)cnt++;
                if(arr[i]>=h+1||i==n){
                    ans+=(cnt*(cnt+1))/2;
                cnt=0;
                }
            }
            cout << ans << '\n';
        }
    }

    return 0;
}

• Subtask 1: 3 / 3
• Subtask 2: 10 / 10
• Subtask 3: 12 / 12
• Subtask 4: 15 / 15
• Subtask 5: 5 / 5
• Subtask 6: 9 / 9
• Subtask 7: 14 / 14
• Subtask 8: 10 / 10
• Subtask 9: 0 / 11
• Subtask 10: 0 / 11