Submission #968089

# Submission time Handle Problem Language Result Execution time Memory
968089 2024-04-23T07:37:22 Z vjudge1 Vudu (COCI15_vudu) C++17
140 / 140
279 ms 20032 KB
#include<bits/stdc++.h>
 
using namespace std;
 
int n, fw[1000001];
long long ans;
 
void update(int idx, int val){
    for(; idx <= n; idx+=(idx&-idx)) fw[idx]+=val;
}
 
long long query(int idx){
    long long sum = 0;
    for(; idx; idx-=(idx&-idx)) sum+=fw[idx];
    return sum;
}
 
int main(){
    cin.tie(nullptr)->sync_with_stdio(false);
    cin >> n;
    long long qs[n+1], k;
    vector<long long> v;
    memset(qs, 0, sizeof(qs));
    for(int i = 1; i<=n; ++i) cin >> qs[i];
    cin >> k;
    for(int i = 1; i<=n; ++i) qs[i]+=qs[i - 1];
    for(int i = 1; i<=n; ++i) v.push_back(qs[i] - i*k);
    sort(v.begin(), v.end());
    v.erase(unique(v.begin(), v.end()), v.end());
    for(int i = 0; i<=n; ++i){
        int id = lower_bound(v.begin(), v.end(), qs[i] - i*k) - v.begin();
        ans+=query(id+1);
        update(id+1, 1);
    }
    cout << ans;
}
# Verdict Execution time Memory Grader output
1 Correct 2 ms 600 KB Output is correct
2 Correct 2 ms 600 KB Output is correct
3 Correct 1 ms 600 KB Output is correct
4 Correct 258 ms 19384 KB Output is correct
5 Correct 146 ms 14792 KB Output is correct
6 Correct 229 ms 18516 KB Output is correct
7 Correct 222 ms 18528 KB Output is correct
8 Correct 195 ms 16068 KB Output is correct
9 Correct 279 ms 20032 KB Output is correct
10 Correct 227 ms 18264 KB Output is correct