#include<bits/stdc++.h>
#define endl "\n"
#define lli long long
using namespace std;
const lli MAXN=1000005;
lli N, a[MAXN], P, pref[MAXN], tree[MAXN], ss[MAXN], ans;
unordered_map<lli, lli> m;
void update(lli i, lli d){
while(i<=N+1){
tree[i]+=d;
i+=i&(-i);
}
return;
}
lli query(lli i){
lli otg=0;
while(i>0){
otg+=tree[i];
i-=i&(-i);
}
return otg;
}
int main() {
ios_base::sync_with_stdio(0);
cout.tie(0);
cin.tie(0);
cin>>N;
for(lli i=1; i<=N; i++){
cin>>a[i];
}
cin>>P;
for(lli i=1; i<=N; i++){
a[i]-=P;
pref[i]=pref[i-1]+a[i];
ss[i]=pref[i];
}
sort(ss, ss+N+1);
lli k=1;
for(lli i=0; i<=N; i++){
if(i!=0 and ss[i]==ss[i-1])continue;
m[ss[i]]=k;
k++;
}
/*update(m[0], 1);
for(lli i=1; i<=N; i++){
ans+=query(m[pref[i]]);
update(m[pref[i]], 1);
}*/
cout<<ans<<endl;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
2 ms |
724 KB |
Output isn't correct |
2 |
Incorrect |
1 ms |
724 KB |
Output isn't correct |
3 |
Incorrect |
1 ms |
596 KB |
Output isn't correct |
4 |
Incorrect |
452 ms |
64560 KB |
Output isn't correct |
5 |
Incorrect |
180 ms |
35864 KB |
Output isn't correct |
6 |
Incorrect |
371 ms |
59636 KB |
Output isn't correct |
7 |
Incorrect |
377 ms |
60472 KB |
Output isn't correct |
8 |
Incorrect |
360 ms |
57576 KB |
Output isn't correct |
9 |
Runtime error |
435 ms |
65536 KB |
Execution killed with signal 9 |
10 |
Incorrect |
399 ms |
59928 KB |
Output isn't correct |