Above the Median

 Above the Median

农夫约翰把他的N（1 <= N <= 100,000）奶牛排在一排来衡量他们的高度，牛i有：高度H_I（1 <= H_I <= 1,000, 
000,000）纳米 – 因为FJ认为他需要精确测量！他想选择一些连续的奶牛拍一张照片发给牛摄影大赛。大赛有一 
个很奇怪的规则，对所有提交的照片：照片有效当且仅当，它描绘了一群中位身高至少大于一定的阈值X（1 <= x  
<= 1,000,000,000）的奶牛。中位身高定义为：有n头奶牛按从小到大顺序排好，第[(1+n)/2]（取上限）头奶牛的身高 
。例如{7，3，2，6}的中位数是6，和{5,4，8}的中位数是5。FJ想知道他有多少种选择。 

输入

*第1行：两个用空格隔开的整数：N和X  
*第2 .. N 1：第i行1包含单个整数H_I。  

输出

*第1行：选择的个数，注意，该数可能超出32位整数的存储范围。

样例

输入

4 6 
10 
5 
6 
2 

输出

7 

提示

有10个可能选择。其中，只有7 个的中位数大于6。它们是{10}，{6}，{10，5}，{5,6}，{6,2}，{10， 5，6}，{1 
0，5，6，2}

#include<bits/stdc++.h>
#define ll long long
#define lowbit(x) x&(-x)
using namespace std;
ll sum[1000001],n,f[1000001],ans,a[1000001],k;
void add(ll x,ll val) {
    while(x<=2*n+1) {
        sum[x]+=val;
        x+=lowbit(x);
    }
}
ll ask(ll x) {
    ll ans=0;
    while(x) {
        ans+=sum[x];
        x-=lowbit(x);
    }
    return ans;
}
int main() {
    scanf("%lld%lld",&n,&k);
    for(ll i=1; i<=n; i++) {
        scanf("%lld",&a[i]);
        if(a[i]>=k)
            f[i]=f[i-1]+1;
        else
            f[i]=f[i-1]-1;
    }
    add(n+1,1);
    for(ll i=1; i<=n; i++) {
        ans+=ask(f[i]+1+n);
        add(f[i]+1+n,1);
    }
    printf("%lld\n",ans);
    return 0;
}