Submission #681071

# Submission time Handle Problem Language Result Execution time Memory
681071 2023-01-12T10:20:43 Z sudheerays123 Vudu (COCI15_vudu) C++17
42 / 140
374 ms 65536 KB
#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 = 1e6+5 , INF = 1e9 , MOD = 1e9+7;

vector<ll> fenwick(N),a(N);

void update(ll node , ll val){
	for(ll i = node; i < N; i += (i & -i)){
		fenwick[i] += val;
	}
}

ll query(ll node){
	ll ans = 0;
	for(ll i = node; i >= 1; i -= (i & -i)){
		ans += fenwick[i];
	}
	return ans;
}
 
void solve(){
 
	ll n;
	cin >> n;
 
	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;
	}

	update(cc[0],1);
 
	ll ans = 0;
	sum = 0;
 
	for(ll i = 1; i <= n; i++){
 
		sum += a[i];
		ll s = sum-p*i;
 
		ans += query(cc[s]);
		update(cc[s],1);
	}
 
	cout << ans;
}	
 
int main(){
 
	fast;
 
	ll tc = 1;
	// cin >> tc;
	while(tc--) solve();
 
	return 0;
}
# Verdict Execution time Memory Grader output
1 Correct 11 ms 16724 KB Output is correct
2 Correct 10 ms 16500 KB Output is correct
3 Correct 11 ms 16496 KB Output is correct
4 Runtime error 374 ms 65536 KB Execution killed with signal 9
5 Runtime error 259 ms 65536 KB Execution killed with signal 9
6 Runtime error 293 ms 65536 KB Execution killed with signal 9
7 Runtime error 273 ms 65536 KB Execution killed with signal 9
8 Runtime error 285 ms 65536 KB Execution killed with signal 9
9 Runtime error 302 ms 65536 KB Execution killed with signal 9
10 Runtime error 308 ms 65536 KB Execution killed with signal 9