Submission #165482

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
165482	2019-11-27T11:23:19 Z	egekabas	Vudu (COCI15_vudu)	C++14	42 / 140	877 ms	65540 KB

vudu

#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;
ll a[1000009];
set<ll> s;
map<ll, ll> mpp; ll cur = 1;
ll 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++;
    }
    ll ans = 0;
    add = 0;
    for(ll i = 0; i < n; ++i){
        if(a[i] >= 0){
            ++ans;
        }
        ans += get(mpp[-a[i]+add]);
        add -= a[i];
        upd(mpp[a[i]+add]);
        
    }
    printf("%lld", ans);
}

Compilation message

vudu.cpp: In function 'int main()':
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);
     ~~~~~^~~~~~~~~~~~

Subtask #1

42.0 / 140.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	9 ms	1272 KB	Output is correct
2	Correct	7 ms	1016 KB	Output is correct
3	Correct	7 ms	1020 KB	Output is correct
4	Runtime error	877 ms	65536 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
5	Runtime error	582 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
6	Runtime error	708 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
7	Runtime error	662 ms	65536 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
8	Runtime error	667 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
9	Runtime error	726 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)
10	Runtime error	716 ms	65540 KB	Execution killed with signal 9 (could be triggered by violating memory limits)