답안 #681043

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
681043 2023-01-12T09:44:01 Z sudheerays123 Vudu (COCI15_vudu) C++17
42 / 140
219 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 = 1e18 , MOD = 1e9+7;

vector<ll> seg(4*N);

void 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];
}

void 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];
}

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;
}

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;
	}

	build(1,1,cnt);
	update(1,1,cnt,cc[0]);

	ll ans = 0;
	sum = 0;

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

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

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

	cout << ans;
}

int main(){

	fast;

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

	return 0;
}
# 결과 실행 시간 메모리 Grader output
1 Correct 19 ms 32340 KB Output is correct
2 Correct 19 ms 32256 KB Output is correct
3 Correct 18 ms 32212 KB Output is correct
4 Runtime error 219 ms 65536 KB Execution killed with signal 9
5 Runtime error 177 ms 65536 KB Execution killed with signal 9
6 Runtime error 219 ms 65536 KB Execution killed with signal 9
7 Runtime error 186 ms 65536 KB Execution killed with signal 9
8 Runtime error 184 ms 65536 KB Execution killed with signal 9
9 Runtime error 201 ms 65536 KB Execution killed with signal 9
10 Runtime error 212 ms 65536 KB Execution killed with signal 9