Submission #875546

#TimeUsernameProblemLanguageResultExecution timeMemory
87554612345678Vudu (COCI15_vudu)C++17
140 / 140
310 ms20000 KiB
#include <bits/stdc++.h>

using namespace std;

const int nx=1e6+5;
long long n, p, qs[nx], res;
vector<long long> v;

struct fenwick
{
    int d[nx];
    void add(int idx)
    {
        for (int i=idx; i<=n; i+=(i&-i)) d[i]++;
    }
    int query(int idx)
    {
        int res=0;
        for (int i=idx; i>0; i-=(i&-i)) res+=d[i];
        return res;
    }
} f;

int main()
{
    cin.tie(NULL)->sync_with_stdio(false);
    cin>>n;
    for (int i=1; i<=n; i++) cin>>qs[i], qs[i]+=qs[i-1];
    cin>>p;
    for (int i=0; i<n; i++) v.push_back(qs[i]-i*p);
    v.push_back(LLONG_MAX);
    sort(v.begin(), v.end());
    //for (auto x:v) cout<<x<<'\n';
    for (int i=1; i<=n; i++)
    {
        f.add(upper_bound(v.begin(), v.end(), qs[i-1]-(i-1)*p)-v.begin());
        res+=f.query(upper_bound(v.begin(), v.end(), qs[i]-(i)*p)-v.begin());
    }
    cout<<res;
}
#Verdict Execution timeMemoryGrader output
Fetching results...