Submission #253617

#	Time	Username	Problem	Language	Result	Execution time	Memory
253617		Vimmer	Vudu (COCI15_vudu)	C++14	42 / 140	688 ms	65540 KiB

vudu

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>

//#pragma GCC optimize("unroll-loops")
//#pragma GCC optimize("-O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC optimize("fast-math")
//#pragma GCC optimize("no-stack-protector")

#define F first
#define S second
#define sz(x) int(x.size())
#define pb push_back
#define pf push_front
#define N 10005
#define M ll(998244353)
#define inf 1e9 + 1e9

using namespace std;
using namespace __gnu_pbds;

typedef long double ld;
typedef long long ll;
typedef unsigned long long ull;
typedef short int si;
typedef array <ll, 3> a3;
typedef array <ll, 4> a4;

typedef tree <ll, null_type, less_equal<ll>, rb_tree_tag, tree_order_statistics_node_update> ordered_set;



ordered_set se, st;

int main()
{
    //freopen("input.txt", "r", stdin); //freopen("output4.txt", "w", stdout);

    ios_base::sync_with_stdio(0); istream::sync_with_stdio(0); cin.tie(0); cout.tie(0);

    ll n;

    cin >> n;

    ll a[n];

    for (ll i = 0; i < n; i++) cin >> a[i];

    ll sum = 0;

    ll p;

    cin >> p;

    for (int i = 0; i < n; i++)
    {
        a[i] -= p;

        sum += a[i];

        se.insert(sum);
    }

    sum = 0;

    ll ans = 0;

    for (int i = 0; i < n; i++)
    {
        ll need = -a[i];

        sum += a[i];

        need += sum;

        ans += sz(se) - se.order_of_key(need);

        ans -= sz(st) - st.order_of_key(need);

        st.insert(sum);
    }

    cout << ans << endl;
}

• Subtask 1: 42 / 140