#include <bits/stdc++.h>
#include <ext/pb_ds/tree_policy.hpp>
#include <ext/pb_ds/assoc_container.hpp>
#define ll long long
using namespace std;
namespace __gnu_pbds
{
typedef tree<ll,
null_type,
less_equal<ll>,
rb_tree_tag,
tree_order_statistics_node_update> ordered_set;
}
using namespace __gnu_pbds;
#define fi ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
//#define ll double
#define ll1 long long
#define F first
#define S second
#define sz size()
#define all(s) s.begin(),s.end()
#define all1(s) s.rbegin(),s.rend()
ll a[1000006];
int arr[1000006];
int main()
{
//freopen("stdin.in","r",stdin);freopen("stdout.out","w",stdout);
int T=1;
//cin>>T;ll oo=0;
while(T--)
{
int n;
cin>>n;
for(int i=0; i<n; i++)
{
cin>>arr[i];
}
int x;
cin>>x;
ordered_set s;
ll c=0;
for(int i=0; i<n; i++)
{
a[i]+=(arr[i]-x);
if(i)a[i]+=a[i-1];
// cout<<c<<" ";
ll k=a[i];
if(k>=0)
{
c+=(i+1-(s.size()-s.order_of_key(k+1)));
}
if(k<0)
{
c+=(s.order_of_key(k+1));
}
s.insert(a[i]);
// cout<<c<<"\n";
}
cout<<c;
}
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
856 KB |
Output is correct |
2 |
Correct |
3 ms |
860 KB |
Output is correct |
3 |
Correct |
3 ms |
860 KB |
Output is correct |
4 |
Runtime error |
671 ms |
65536 KB |
Execution killed with signal 9 |
5 |
Correct |
346 ms |
41040 KB |
Output is correct |
6 |
Correct |
570 ms |
64588 KB |
Output is correct |
7 |
Runtime error |
532 ms |
65536 KB |
Execution killed with signal 9 |
8 |
Correct |
509 ms |
58444 KB |
Output is correct |
9 |
Runtime error |
572 ms |
65536 KB |
Execution killed with signal 9 |
10 |
Correct |
567 ms |
65216 KB |
Output is correct |