oj.uz

Submission #762738

# Submission time Handle Problem Language Result Execution time Memory
762738 2023-06-21T17:20:32 Z ETheBest3 Vudu (COCI15_vudu) C++14
0 / 140
 328 ms 65536 KB 
#include<bits/stdc++.h>
#define endl "\n"
#define lli long long
using namespace std;
const lli MAXN=2000005;
lli N, a[MAXN], P, pref[MAXN], tree[MAXN], ss[MAXN], ans;
map<lli, lli> m;
void update(lli i, lli d){
    while(i<=N+1){
        tree[i]+=d;
        i+=i&(-i);
    }
    return;
}
lli query(lli i){
    lli otg=0;
    while(i>0){
        otg+=tree[i];
        i-=i&(-i);
    }
    return otg;
}
int main() {
    ios_base::sync_with_stdio(0);
    cout.tie(0);
    cin.tie(0);
    cin>>N;
    for(lli i=1; i<=N; i++){
        cin>>a[i];
    }
    cin>>P;
    for(lli i=1; i<=N; i++){
        a[i]-=P;
        pref[i]=pref[i-1]+a[i];
        ss[i]=pref[i];
    }
    //sort(ss, ss+N+1);
    lli k=1;
    for(lli i=0; i<=N; i++){
        if(i==0){
            m[ss[i]]=k;
            k++;
        }
        if(i!=0 and ss[i]==ss[i-1])continue;
        m[ss[i]]=k;
        k++;
    }
    update(m[0], 1);
    for(lli i=1; i<=N; i++){
        ans+=query(m[pref[i]]);
        update(m[pref[i]], 1);
    }
    cout<<ans<<endl;
    return 0;
}

Subtask #1
0 / 140.0

# Verdict Execution time Memory Grader output
1 Incorrect 3 ms 980 KB Output isn't correct
2 Incorrect 2 ms 852 KB Output isn't correct
3 Incorrect 2 ms 724 KB Output isn't correct
4 Runtime error 294 ms 65536 KB Execution killed with signal 9
5 Incorrect 328 ms 51696 KB Output isn't correct
6 Runtime error 225 ms 65536 KB Execution killed with signal 9
7 Runtime error 237 ms 65536 KB Execution killed with signal 9
8 Runtime error 219 ms 65536 KB Execution killed with signal 9
9 Runtime error 227 ms 65536 KB Execution killed with signal 9
10 Runtime error 238 ms 65536 KB Execution killed with signal 9