-
Notifications
You must be signed in to change notification settings - Fork 10
/
KMEDIAN.cpp
63 lines (51 loc) 路 1.24 KB
/
KMEDIAN.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> ii;
typedef unsigned long long ull;
#define X first
#define Y second
#define pb push_back
#define mp make_pair
#define ep emplace_back
#define EL printf("\n")
#define sz(A) (int) A.size()
#define FOR(i,l,r) for (int i=l;i<=r;i++)
#define FOD(i,r,l) for (int i=r;i>=l;i--)
#define fillchar(a,x) memset(a, x, sizeof (a))
#define faster ios_base::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);
const int N = 500000;
int n, k, a[N], f[N], b[N];
ll t[N], res;
void update(int x) {
for (x; x<=a[0]; x+=x&-x) t[x]++;
}
ll get(int x) {
ll ans = 0;
for (x; x>0; x-=x&-x) ans += t[x];
return ans;
}
int main() {
// freopen("INP.TXT", "r", stdin);
// freopen("OUT.TXT", "w", stdout);
scanf("%d%d", &n,&k);
for (int i=1; i<=n; i++) {
scanf("%d", &a[i]);
f[i] = f[i-1] + (a[i] >= k);
b[++b[0]] = 2*f[i]-i-1;
b[++b[0]] = 2*f[i-1]-i;
}
sort(b+1,b+b[0]+1);
b[++b[0]] = N;
for (int i=1; i<b[0]; i++)
if (b[i] < b[i+1]) a[++a[0]] = b[i];
for (int j=1; j<=n; j++) {
int x;
x = lower_bound(a+1,a+a[0]+1,2*f[j-1]-j)-a;
update(x);
x = lower_bound(a+1,a+a[0]+1,2*f[j]-j-1)-a;
res += get(x);
}
printf("%lld", res);
return 0;
}