oj.uz

Submission #977003

# Submission time Handle Problem Language Result Execution time Memory
977003 2024-05-07T10:15:38 Z SeenSiravit Vudu (COCI15_vudu) C++14
140 / 140
 220 ms 55012 KB 
#include<bits/stdc++.h>
#define ll long long

using namespace std;

const int mxN = 1e6 + 5;

struct DATA
{
    int idx;
    ll val;
};


int n;
ll a[mxN];
DATA b[mxN];
ll p;
int fw[mxN];
vector<DATA> tmp;
int mp[mxN];

bool cmp(DATA x , DATA y){
    if(x.val == y.val) return x.idx < y.idx;
    return x.val < y.val;
}

void update(int idx , int val){
    for(int i=idx;i<mxN;i += (i & -i)) fw[i] += val;
}

ll query(int idx){
    ll res = 0;
    for(int i=idx;i>0;i -= (i & -i)) res += fw[i];
    return res;
}

int main(){
    ios::sync_with_stdio(0),cin.tie(0);
    
    cin>> n;

    for(int i=1;i<=n;i++) cin>> a[i] , a[i] += a[i-1];

    // for(int i=1;i<=n;i++) cout<< a[i] << " ";
    // cout<< "\n";

    cin>> p;

    tmp.push_back({0,0});
    for(int i=1;i<=n;i++) b[i].val = a[i] - p*i , b[i].idx = i , tmp.push_back(b[i]);
    // for(int i=1;i<=n;i++) cout<< b[i] << " ";
    // cout<< "\n\n";

    

    sort(tmp.begin() , tmp.end() , cmp);

    // for(auto elem : tmp) cout<< elem.idx << " " << elem.val << " , ";
    // cout<< "\n";

    int id = 1;
    for(auto elem : tmp){
        mp[elem.idx] = id;
        id++;
    }

    ll ans = 0;

    update(mp[0] , 1);

    for(int i=1;i<=n;i++){
        int cnt = query(mp[b[i].idx]);

        // cout<< "\ncnt = " << cnt << " ";

        // cout<< cnt << "\n\n";

        ans += cnt;

        update(mp[i] , 1);
    }

    cout<< ans;

    return 0;
}

Subtask #1
140.0 / 140.0

# Verdict Execution time Memory Grader output
1 Correct 3 ms 8796 KB Output is correct
2 Correct 3 ms 8796 KB Output is correct
3 Correct 2 ms 8896 KB Output is correct
4 Correct 220 ms 53936 KB Output is correct
5 Correct 123 ms 37176 KB Output is correct
6 Correct 186 ms 52404 KB Output is correct
7 Correct 189 ms 53588 KB Output is correct
8 Correct 167 ms 49380 KB Output is correct
9 Correct 218 ms 55012 KB Output is correct
10 Correct 184 ms 52400 KB Output is correct