#include <bits/stdc++.h>
using namespace std;
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
using namespace __gnu_pbds;
#define ordered_set tree<ll, null_type, less_equal<ll>, rb_tree_tag, tree_order_statistics_node_update>
#define nn "\n"
#define x_x ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0);
#define intt int _; cin >> _; while (_--)
#define emp push_back
#define mod 1000000007
#define all(v) v.begin(), v.end()
#define ld long double
#define A first
#define B second
//#define int long long
typedef long long ll;
const ld eps = 1e-27;
// diff between decimals 0.000000001 mt19937 mt(time(nullptr));
int main() {
x_x
int n; cin>>n; ll p, sm=0, ar[n];
for(auto&i:ar)cin>>i;
cin>>p; ordered_set s;
ll ans=0; s.insert(ans);
for (ll i=1; i<=n; i++) {
sm+=ar[i-1]; ll x=p*i; x=sm-x;
ans+=s.order_of_key(x+1); x=-p*i; x+=sm; s.insert(x);
}
// s.insert(1); s.insert(2); s.insert(3); cout<<s.order_of_key(5);
cout<<ans;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
3 ms |
860 KB |
Output is correct |
2 |
Correct |
2 ms |
860 KB |
Output is correct |
3 |
Correct |
2 ms |
860 KB |
Output is correct |
4 |
Runtime error |
451 ms |
65536 KB |
Execution killed with signal 9 |
5 |
Correct |
242 ms |
38736 KB |
Output is correct |
6 |
Correct |
430 ms |
60656 KB |
Output is correct |
7 |
Correct |
402 ms |
62900 KB |
Output is correct |
8 |
Correct |
362 ms |
55188 KB |
Output is correct |
9 |
Runtime error |
388 ms |
65536 KB |
Execution killed with signal 9 |
10 |
Correct |
406 ms |
61508 KB |
Output is correct |