/*
Editor: Abdelrahman Hossam
Nickname: Blobo2_Blobo2
IOI next year isA :)
*/
/*#pragma GCC optimize("Ofast,no-stack-protector,unroll-loops,fast-math,O3")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,sse4.2,popcnt,abm,mmx,avx2,tune=native")
#pragma GCC optimize("-ffast-math")
#pragma GCC optimize("-funroll-loops")
#pragma GCC optimize("-funroll-all-loops,-fpeel-loops,-funswitch-loops")*/
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define endl "\n"
#define all(v) v.begin(),v.end()
#define gen(arr,n,nxt) generate(arr,arr+n,nxt)
#define Blobo2_el_7akim_elly_3ayz_yro7_IOI_w_3ayz_yakol_jilaty ios_base::sync_with_stdio(false);cin.tie(0);
const int mo=1e9+7,INF=1e18;
int nxt(){int x;cin>>x;return x;}
int tree[4000005];
int arr[1000000],n,p;
void update(int pos,int val,int idx=1,int ss=0,int se=n-1){
if(ss==se){
tree[idx]+=val;
return;
}
int mid=(ss+se)>>1;
if(pos<=mid)update(pos,val,idx<<1,ss,mid);
else update(pos,val,idx<<1|1,mid+1,se);
tree[idx] = tree[idx<<1]+tree[idx<<1|1];
}
int sum(int l,int r,int idx=1,int ss=0,int se=n-1){
if(l>se||r<ss)
return 0;
if(l<=ss&&se<=r)
return tree[idx];
int mid=(ss+se)>>1;
return sum(l,r,idx<<1,ss,mid)+
sum(l,r,idx<<1|1,mid+1,se);
}
signed main(){
Blobo2_el_7akim_elly_3ayz_yro7_IOI_w_3ayz_yakol_jilaty
n=nxt();
gen(arr,n,nxt);
p=nxt();
for(int i=0;i<n;i++)arr[i]-=p;
int c=0,idx=0;
vector<pair<int,int> >v;
for(int i=0;i<n;i++){
c+=arr[i];
v.push_back({c,i});
}
sort(all(v));
int mp[n+1];
for(int i=0;i<v.size();i++){
if(i==0)
mp[v[i].second]=++idx;
else if(v[i].first!=v[i-1].first)
mp[v[i].second]=++idx;
else
mp[v[i].second]=idx;
}
int cnt=0,ans=0;
for(int i=0;i<n;i++){
cnt+=arr[i];
ans+=sum(1,mp[i]);
update(mp[i],1);
if(cnt>=0)ans++;
}
cout<<ans;
return 0;
}
Compilation message
vudu.cpp: In function 'int main()':
vudu.cpp:56:18: warning: comparison of integer expressions of different signedness: 'long long int' and 'std::vector<std::pair<long long int, long long int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
56 | for(int i=0;i<v.size();i++){
| ~^~~~~~~~~
# |
Verdict |
Execution time |
Memory |
Grader output |
1 |
Correct |
5 ms |
716 KB |
Output is correct |
2 |
Correct |
3 ms |
716 KB |
Output is correct |
3 |
Correct |
3 ms |
716 KB |
Output is correct |
4 |
Correct |
832 ms |
47120 KB |
Output is correct |
5 |
Correct |
425 ms |
33948 KB |
Output is correct |
6 |
Incorrect |
699 ms |
43776 KB |
Output isn't correct |
7 |
Correct |
705 ms |
44716 KB |
Output is correct |
8 |
Correct |
615 ms |
40936 KB |
Output is correct |
9 |
Correct |
786 ms |
48096 KB |
Output is correct |
10 |
Correct |
695 ms |
43920 KB |
Output is correct |