Submission #968646

# Submission time Handle Problem Language Result Execution time Memory
968646 2024-04-23T19:25:21 Z VMaksimoski008 Vudu (COCI15_vudu) C++14
112 / 140
663 ms 65536 KB
#include <bits/stdc++.h>
#define all(x) x.begin(), x.end()
using namespace std;
using ll = long long;
#define int long long

struct BIT {
    int n;
    vector<ll> tree;

    void config(int _n) {
        n = _n + 10;
        tree.resize(_n+60);
    }

    void update(int p, int v) {
        for(p++; p<n; p+=p&-p) tree[p] += v;
    }

    ll query(int p) {
        ll ans = 0;
        for(p++; p>0; p-=p&-p) ans += tree[p];
        return ans;
    }
};


int32_t main() {
    int n, p;
    ll ans = 0;
    cin >> n;

    vector<ll> v(n+1);
    for(int i=1; i<=n; i++) cin >> v[i];

    cin >> p;
    ll sum = 0;
    for(int i=1; i<=n; i++) {
        sum += v[i];
        v[i] = sum - i * p;
    }

    set<ll> s;
    for(ll &x : v) s.insert(x);
    vector<int> comp(all(s));
    for(ll &x : v) x = lower_bound(all(comp), x) - comp.begin();

    BIT bit; bit.config(n);
    bit.update(v[0], 1);

    for(int i=1; i<=n; i++) {
        ans += bit.query(v[i]);
        bit.update(v[i], 1);
    }

    cout << ans << '\n';
    return 0;
}
# Verdict Execution time Memory Grader output
1 Correct 3 ms 856 KB Output is correct
2 Correct 2 ms 600 KB Output is correct
3 Correct 2 ms 604 KB Output is correct
4 Runtime error 663 ms 65536 KB Execution killed with signal 9
5 Correct 355 ms 38752 KB Output is correct
6 Correct 594 ms 60620 KB Output is correct
7 Correct 588 ms 65508 KB Output is correct
8 Correct 517 ms 54796 KB Output is correct
9 Runtime error 586 ms 65536 KB Execution killed with signal 9
10 Correct 579 ms 62252 KB Output is correct