oj.uz

답안 #219211

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
219211 2020-04-04T15:17:01 Z DavidDamian Vudu (COCI15_vudu) C++11
56 / 140
 510 ms 65540 KB 
#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
ll bit[1000006];
int n;
void update(int i,int v)
{
    while(i<=n){
        bit[i]+=v;
        i+=(i&(-i));
    }
}
int query(int i)
{
    int ans=0;
    while(i>0){
        ans+=bit[i];
        i-=(i&(-i));
    }
    return ans;
}
int getSum(int a,int b)
{
    return query(b)-query(a-1);
}
typedef pair<ll,int> pii;
map<ll,int> remap;
ll k;
ll A[1000006];
ll S[1000006];
ll S_K[1000006]; //S[i]-k*i
int main()
{
    ios_base::sync_with_stdio(0);cin.tie(0);
    cin>>n;
    for(int i=1;i<=n;i++){
        cin>>A[i];
    }
    cin>>k;
    remap[0]=1;
    for(ll i=1;i<=n;i++){
        S[i]=S[i-1]+A[i];
        S_K[i]=S[i]-k*i;
        remap[ S_K[i] ]=1;
    }
    int id=1;
    for(pii p: remap){
        remap[p.first]=id++;
    }
    ll total=0;
    for(int i=0;i<=n;i++){
        S_K[i]=remap[ S_K[i] ];
        if(i>0) total+=getSum(1,S_K[i]);
        update(S_K[i],1);
    }
    cout<<total<<'\n';
    return 0;
}

Subtask #1
56.0 / 140.0

# 결과 실행 시간 메모리 Grader output
1 Correct 8 ms 1024 KB Output is correct
2 Correct 7 ms 896 KB Output is correct
3 Correct 7 ms 896 KB Output is correct
4 Runtime error 452 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
5 Correct 510 ms 51964 KB Output is correct
6 Runtime error 370 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
7 Runtime error 351 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
8 Runtime error 348 ms 65536 KB Execution killed with signal 9 (could be triggered by violating memory limits)
9 Runtime error 389 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
10 Runtime error 370 ms 65536 KB Execution killed with signal 9 (could be triggered by violating memory limits)