Submission #681038

#	Time	Username	Problem	Language	Result	Execution time	Memory
681038		sudheerays123	Vudu (COCI15_vudu)	C++17	42 / 140	422 ms	65536 KiB

vudu

#include<bits/stdc++.h>
using namespace std;
#define fast ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
#define ll long long int
const ll N = 100+5 , INF = 1e18 , MOD = 1e9+7;

void solve(){

	ll n;
	cin >> n;

	vector<ll> a(n+5);
	for(ll i = 1; i <= n; i++) cin >> a[i];

	ll p;
	cin >> p;

	ll sum = 0;
	set<ll> val;

	val.insert(0);
	for(ll i = 1; i <= n; i++){
		sum += a[i];
		val.insert(sum-p*i);
	}

	ll cnt = 0;
	map<ll,ll> cc;
	for(auto u : val){
		cnt++;
		cc[u] = cnt;
	}

	vector<ll> seg(4*cnt);

	function<void(ll,ll,ll)> build = [&](ll node , ll left , ll right){

		if(left == right){
			seg[node] = 0;
			return;
		}

		ll mid = (left+right)/2;

		build(2*node,left,mid);
		build(2*node+1,mid+1,right);

		seg[node] = seg[2*node]+seg[2*node+1];
	};

	build(1,1,cnt);

	function<void(ll,ll,ll,ll)> update = [&](ll node , ll left , ll right , ll pos){

		if(left == right){
			seg[node]++;
			return;
		}

		ll mid = (left+right)/2;

		if(pos <= mid) update(2*node,left,mid,pos);
		else update(2*node+1,mid+1,right,pos);

		seg[node] = seg[2*node]+seg[2*node+1];
	};

	update(1,1,n,cc[0]);

	function<ll(ll,ll,ll,ll,ll)> query = [&](ll node , ll left , ll right , ll ql , ll qr){

		if(left >= ql && right <= qr) return seg[node];
		if(left > qr || right < ql) return 0ll;

		ll mid = (left+right)/2;

		ll l = query(2*node,left,mid,ql,qr);
		ll r = query(2*node+1,mid+1,right,ql,qr);

		return l+r;
	};

	ll ans = 0;
	sum = 0;

	for(ll i = 1; i <= n; i++){

		sum += a[i];
		ll s = sum-p*i;

		ans += query(1,1,n,1,cc[s]);
		update(1,1,n,cc[s]);
	}

	cout << ans;
}	

int main(){

	fast;

	ll tc = 1;
	// cin >> tc;
	while(tc--) solve();

	return 0;
}

• Subtask 1: 42 / 140