Submission #232617

# Submission time Handle Problem Language Result Execution time Memory
232617 2020-05-17T16:30:22 Z soroush Vudu (COCI15_vudu) C++14
42 / 140
1000 ms 61336 KB
#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef pair<ll  ,ll > pii;

typedef long double ld;

mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());

const ll maxn  = 1e6+100;
const ll mod =1e9+7;
const ld PI = acos((ld)-1);

#define pb(x) push_back(x);
#define endl '\n'
#define dokme(x) return(cout << x , 0);
#define migmig ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define ms(x , y) memset(x , y , sizeof x);
#define file_init freopen("input.txt", "r+", stdin); freopen("output.txt", "w+", stdout);
ll pw(ll a, ll b, ll md = mod){ll res = 1;while(b){if(b&1){res=(a*res)%md;}a=(a*a)%md;b>>=1;}return(res);}

int n , a[maxn] , p , ans;
int fen[maxn];
vector < int > vec;
map < int , int > mp;

void update(int x){
	for(;x<=n ;x+=x&-x)
		fen[x]++;
}

int get(int pos){
	int ans = 0;
	for(;pos;pos-=pos&-pos)
		ans+=fen[pos];
	return(ans);
}

int main(){
    migmig
	cin >> n;
	for(int i = 1 ; i <= n ; i ++)
		cin >> a[i];
	cin >> p;
	for(int i = 1 ; i <= n ; i ++)
		a[i]-=p;
	ll sum = 0;
	for(int i = 1 ; i <= n ; i ++){
		sum+=a[i];
		vec.pb(sum);
	}
	sort(vec.begin() , vec.end());
	int cur = 0;
	for(auto i : vec){
		if(mp[i] == 0)mp[i] = ++cur;
	}
	swap(n , cur);
	sum = 0;
	for(int i = 1 ; i <= cur ; i ++){
		sum+=a[i];
		ans+=get(mp[sum]) + (sum>=0);
		update(mp[sum]);
	}
	cout << ans;
    return(0);
}
# Verdict Execution time Memory Grader output
1 Correct 9 ms 768 KB Output is correct
2 Correct 7 ms 640 KB Output is correct
3 Correct 7 ms 768 KB Output is correct
4 Execution timed out 1097 ms 59492 KB Time limit exceeded
5 Incorrect 943 ms 34912 KB Output isn't correct
6 Execution timed out 1100 ms 52988 KB Time limit exceeded
7 Execution timed out 1099 ms 55132 KB Time limit exceeded
8 Execution timed out 1094 ms 48224 KB Time limit exceeded
9 Execution timed out 1093 ms 61336 KB Time limit exceeded
10 Execution timed out 1100 ms 53708 KB Time limit exceeded