Submission #165495

# Submission time Handle Problem Language Result Execution time Memory
165495 2019-11-27T11:35:51 Z egekabas Vudu (COCI15_vudu) C++14
42 / 140
1000 ms 65540 KB
#include <bits/stdc++.h>
#define ff first
#define ss second
#define pb push_back
#define mp make_pair
using namespace std;
typedef long long   ll;
typedef unsigned long long   ull;
typedef long double ld;
typedef pair<ll, ll>    pll;
typedef pair<ull, ull>    pull;
typedef pair<ll, ll>  pii;
typedef pair<ld, ld>  pld;
ll n, p;
int a[1000009];
set<ll> s;
map<ll, int> mpp; ll cur = 1;
int bit[1000009];
void upd(ll idx){
    while(idx > 0){
        bit[idx]++;
        idx -= idx&(-idx);
    }
}
ll get(ll idx){
    ll ret = 0;
    while(idx <= cur+2){
        ret += bit[idx];
        idx += idx&(-idx);
    }
    return ret;
}
int main() {
    
    //freopen("in.txt", "r", stdin);
//    freopen("out.txt", "w", stdout);
    
    scanf("%lld", &n);
    for(ll i = 0; i < n; ++i){
        scanf("%lld", &a[i]);
    }
    scanf("%lld", &p);
    for(ll i = 0; i < n; ++i){
        a[i] -= p;
    }
    
    ll add = 0;
    for(ll i = 0; i < n; ++i){
        //s.insert(-a[i]+add);
        add -= a[i];
        s.insert(a[i]+add);    
    }
    for(auto u : s){
        mpp[u] = cur++;
    }
    s.clear();
    ll ans = 0;
    add = 0;
    for(ll i = 0; i < n; ++i){
        if(a[i] >= 0){
            ++ans;
        }
        if(mpp.lower_bound(-a[i]+add) != mpp.end())
            ans += get((*mpp.lower_bound(-a[i]+add)).ss);
        add -= a[i];
        upd(mpp[a[i]+add]);
    }
    printf("%lld", ans);
}

Compilation message

vudu.cpp: In function 'int main()':
vudu.cpp:40:28: warning: format '%lld' expects argument of type 'long long int*', but argument 2 has type 'int*' [-Wformat=]
         scanf("%lld", &a[i]);
                       ~~~~~^
vudu.cpp:38:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%lld", &n);
     ~~~~~^~~~~~~~~~~~
vudu.cpp:40:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld", &a[i]);
         ~~~~~^~~~~~~~~~~~~~~
vudu.cpp:42:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
     scanf("%lld", &p);
     ~~~~~^~~~~~~~~~~~
# Verdict Execution time Memory Grader output
1 Correct 9 ms 1144 KB Output is correct
2 Correct 7 ms 892 KB Output is correct
3 Correct 7 ms 888 KB Output is correct
4 Runtime error 751 ms 65536 KB Execution killed with signal 9 (could be triggered by violating memory limits)
5 Execution timed out 1087 ms 64588 KB Time limit exceeded
6 Runtime error 632 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
7 Runtime error 617 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
8 Runtime error 594 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
9 Runtime error 637 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)
10 Runtime error 653 ms 65540 KB Execution killed with signal 9 (could be triggered by violating memory limits)