/*
ID: blackha5
TASK: test
LANG: C++
*/
#include <bits/stdc++.h>
#define ff first
#define ss second
#define Maxn 1000003
#define ll long long
#define pb push_back
#define Inf 1000000009
#define ppb() pop_back()
#define pii pair <int , int>
#define mid(x, y) (x + y) / 2
#define all(x) x.begin(),x.end()
#define llInf 1000000000000000009
using namespace std;
ll ans;
int n, p;
int a[Maxn];
int T[Maxn * 4];
vector <pair <ll, int> > v;
void upd (int x, int l, int r, int v) {
if (l == r) {
T[v] ++;
return;
}
if (x <= mid (l, r))
upd (x, l, mid (l, r), v * 2);
else
upd (x, mid (l, r) + 1, r, v * 2 + 1);
T[v] = T[v * 2] + T[v * 2 + 1];
return;
}
int get (int x, int y, int l, int r, int v) {
if (l > y or r < x)
return 0;
if (l >= x and r <= y)
return T[v];
return get (x, y, l, mid (l, r), v * 2) + get (x, y, mid (l, r) + 1, r, v * 2 + 1);
}
int main () {
//freopen ("file.in", "r", stdin);
//freopen ("file.out", "w", stdout);
//srand ((unsigned) time ( NULL ));
//int randomNumber = rand() % 10 + 1;
scanf ("%d", &n);
for (int i = 1; i <= n; i++)
scanf ("%d", a + i);
scanf ("%d", &p);
for (int i = 1; i <= n; i++)
a[i] -= p;
ll s = 0;
for (int i = 1; i <= n; i++) {
s += a[i];
v.pb ({s, i});
}
sort (all(v));
v.pb ({0, 0});
int c = 0;
for (int i = 0; i < v.size(); i++) {
if (i == 0)
a[v[i].ss] = ++c;
else if (v[i].ff != v[i - 1].ff)
a[v[i].ss] = ++c;
else
a[v[i].ss] = c;
}
upd (a[0], 1, c, 1);
for (int i = 1; i <= n; i++) {
ans += get (1, a[i], 1, c, 1);
upd (a[i], 1, c, 1);
}
printf ("%lld\n", ans);
return 0;
}
Compilation message
vudu.cpp: In function 'int main()':
vudu.cpp:78:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
for (int i = 0; i < v.size(); i++) {
~~^~~~~~~~~~
vudu.cpp:59:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf ("%d", &n);
~~~~~~^~~~~~~~~~
vudu.cpp:62:9: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf ("%d", a + i);
~~~~~~^~~~~~~~~~~~~
vudu.cpp:64:8: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf ("%d", &p);
~~~~~~^~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
6 ms |
760 KB |
Output isn't correct |
2 |
Incorrect |
5 ms |
1004 KB |
Output isn't correct |
3 |
Incorrect |
5 ms |
1004 KB |
Output isn't correct |
4 |
Incorrect |
717 ms |
36620 KB |
Output isn't correct |
5 |
Incorrect |
388 ms |
36620 KB |
Output isn't correct |
6 |
Incorrect |
613 ms |
48044 KB |
Output isn't correct |
7 |
Incorrect |
617 ms |
57160 KB |
Output isn't correct |
8 |
Incorrect |
562 ms |
58740 KB |
Output isn't correct |
9 |
Incorrect |
710 ms |
63272 KB |
Output isn't correct |
10 |
Incorrect |
640 ms |
63272 KB |
Output isn't correct |