oj.uz

Submission #220372

# Submission time Handle Problem Language Result Execution time Memory
220372 2020-04-07T18:43:56 Z XmtosX Vudu (COCI15_vudu) C++17
70 / 140
 802 ms 65540 KB 
#include <bits/stdc++.h>
using namespace std;
const int N=1e6+6;
int n,a[N],p;
long long pfx[N];
set<long long> s;
int BIT[N];
map <long long,int> r;
int query(int x)
{
    int sum = 0;
    x++;
    while (x>0)
    {
        sum+= BIT[x];
        x-=x&(-x);
    }
    return sum;
}
void update(int x)
{
    x++;
    while (x <= n)
    {
        BIT[x] ++;
        x+=x&(-x);
    }
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    scanf("%d",&p);
    for (int i=1;i<=n;i++)
    {
        pfx[i]=pfx[i-1]+a[i]-p;
        s.insert(pfx[i]);
    }
    s.insert(0);
    int cnt=0;
    while (!s.empty())
    {
        r[*s.begin()]=cnt++;
        s.erase(s.begin());
    }
    for (int i=0;i<=n;i++)
    {
        pfx[i]=r[pfx[i]];
    }
    long long ans=0;
    for (int i=0;i<=n;i++)
    {
        ans+= query(pfx[i]);
        update(pfx[i]);
    }
    cout <<ans;
    return 0;
}

Compilation message

vudu.cpp: In function 'int main()':
vudu.cpp:31:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d",&n);
     ~~~~~^~~~~~~~~
vudu.cpp:33:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%d",&a[i]);
         ~~~~~^~~~~~~~~~~~
vudu.cpp:34:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d",&p);
     ~~~~~^~~~~~~~~

Subtask #1
70.0 / 140.0

# Verdict Execution time Memory Grader output
1 Correct 9 ms 896 KB Output is correct
2 Correct 8 ms 768 KB Output is correct
3 Correct 8 ms 768 KB Output is correct
4 Runtime error 612 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
5 Correct 565 ms 45432 KB Output is correct
6 Runtime error 560 ms 65536 KB Execution killed with signal 9 (could be triggered by violating memory limits)
7 Runtime error 539 ms 65536 KB Execution killed with signal 9 (could be triggered by violating memory limits)
8 Correct 802 ms 64376 KB Output is correct
9 Runtime error 522 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
10 Runtime error 567 ms 65536 KB Execution killed with signal 9 (could be triggered by violating memory limits)