Submission #1272127

#TimeUsernameProblemLanguageResultExecution timeMemory
1272127ahmd_ibraaaVudu (COCI15_vudu)C++20
42 / 140
776 ms51192 KiB
#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace std;
using namespace __gnu_pbds;

#define ordered_set tree<int, null_type,less_equal<int>, rb_tree_tag,tree_order_statistics_node_update>
#define ll long long
#define double long double
#define medal ios_base::sync_with_stdio(false);cin.tie(NULL);cout.tie(NULL);
#define fi first
#define sec second
#define endl '\n'

signed main(){
  medal
	int n;
	cin>>n;
	ordered_set ms;
	int a[n+1];
	for(int i=1; i<=n; i++){
		cin>>a[i];
	}
	int p; cin>>p;
	for(int i=1; i<=n; i++){
		a[i]-=p;
	}
	
	ll ans = 0;
	ll pref = 0;
	ms.insert(0);
	
	for(int i=1; i<=n; i++){
		pref += a[i];
		int cnt = ms.order_of_key(pref+1);
		ans += cnt;
		ms.insert(pref);
	}
	cout<<ans;
}
#Verdict Execution timeMemoryGrader output
Fetching results...