#include <bits/stdc++.h>
using namespace std;
typedef long long int ll;
ll bit[1000006];
int n;
void update(int i,int v)
{
while(i<=n){
bit[i]+=v;
i+=(i&(-i));
}
}
int query(int i)
{
int ans=0;
while(i>0){
ans+=bit[i];
i-=(i&(-i));
}
return ans;
}
int getSum(int a,int b)
{
return query(b)-query(a-1);
}
typedef pair<ll,int> pii;
map<ll,int> remap;
ll k;
ll A[1000006];
ll S_K[1000006]; //S[i]-k*i
int main()
{
ios_base::sync_with_stdio(0);cin.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
cin>>A[i];
A[i]+=A[i-1];
}
cin>>k;
remap[0]=1;
for(ll i=1;i<=n;i++){
S_K[i]=A[i]-k*i;
remap[ S_K[i] ]=1;
}
int id=1;
for(pii p: remap){
remap[p.first]=id++;
}
ll total=0;
for(int i=0;i<=n;i++){
S_K[i]=remap[ S_K[i] ];
if(i>0) total+=getSum(1,S_K[i]);
update(S_K[i],1);
}
cout<<total<<'\n';
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
8 ms |
1024 KB |
Output is correct |
2 |
Correct |
7 ms |
768 KB |
Output is correct |
3 |
Correct |
7 ms |
768 KB |
Output is correct |
4 |
Runtime error |
458 ms |
65540 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
5 |
Correct |
513 ms |
47480 KB |
Output is correct |
6 |
Runtime error |
417 ms |
65536 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
7 |
Runtime error |
371 ms |
65536 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
8 |
Runtime error |
496 ms |
65540 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
9 |
Runtime error |
399 ms |
65536 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |
10 |
Runtime error |
388 ms |
65540 KB |
Execution killed with signal 9 (could be triggered by violating memory limits) |