#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
#define MAXN 1000005
ll a[MAXN];
int bit[MAXN];
map <ll,int> mp;
void update(int idx,int val)
{
while(idx < MAXN)
{
bit[idx] += val;
idx += idx&(-idx);
}
}
ll get(int idx)
{
ll ret = 0LL;
while(idx > 0)
{
ret += bit[idx];
idx -= idx&(-idx);
}
return ret;
}
ll get(int l,int r)
{
ll ret = get(r);
if(l > 0)ret -= get(l-1);
return ret;
}
vector <int> all;
int main()
{
ios_base::sync_with_stdio(false);
int n;
cin >> n;
for(int i = 1; i <= n; i++)cin >> a[i];
int p;
cin >> p;
for(int i = 1; i <= n; i++)a[i]-=p,a[i]+=a[i-1];
all.push_back(0);
for(int i = 1; i <= n; i++)all.push_back(a[i]);
sort(all.begin(),all.end());
//all.erase(unique(all.begin(),all.end()),all.end());
all.resize(unique(all.begin(), all.end()) - a.begin());
int ze = -1;
for(int i = 0; i < all.size(); i++)if(all[i]==0){ze=i;break;}
update(ze+1,1);
ll ans = 0LL;
for(int i = 1; i <= n; i++)
{
int idx = lower_bound(all.begin(),all.end(),a[i])-all.begin()+1;
ans += 1LL*get(idx);
update(idx,1);
}
cout << ans;
return 0;
}
Compilation message
vudu.cpp: In function 'int main()':
vudu.cpp:48:49: error: request for member 'begin' in 'a', which is of non-class type 'll [1000005] {aka long long int [1000005]}'
all.resize(unique(all.begin(), all.end()) - a.begin());
^~~~~
vudu.cpp:50:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for(int i = 0; i < all.size(); i++)if(all[i]==0){ze=i;break;}
~~^~~~~~~~~~~~
