【树状数组逆序对】USACO.2011JAN-Above the median

【题意】

给出一串数字，问中位数大于等于X的连续子串有几个。（这里如果有偶数个数，定义为偏大的那一个而非中间取平均）

【思路】

下面的数据规模也小于原题，所以要改成__int64才行。没找到测试数据，自己编的几组。简单来说读入每个数，大于等于中位数设为1，小于设为-1，前i个数的和建立一个树状数组，求逆序对。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int MAXN=10000;
struct Rec
{
       int num,orinum; 
       bool operator < (const Rec &x) const
       {
            return num<x.num;
       }
};
int n,x;
Rec b[MAXN];
int a[MAXN],e[MAXN];
int ans=0;

int lowbit(int x)
{
    return (x & (-x));
}

void add(int p,int x)
{
     int i=p;
     while (i<=(2*n+1))
     {
           e[i]+=x;
           i=i+lowbit(i);
     }
}

int sum(int p)
{
    int s=0;
    int i=p;
    while (i>0)
    {
          s+=e[i];
          i=i-lowbit(i);
    }
    return s;
}

int main()
{
    /*input*/
    scanf("%d%d",&n,&x);
    int num[MAXN];

    a[0]=0;
    for (int i=1;i<=n;i++)
    {
        a[i]=a[i-1];
        scanf("%d",&num[i]);
        if (num[i]>=x) a[i]++;
        if (num[i]<x) a[i]--;
    }
    
    
    add(n+1,1);
    for (int i=1;i<=n;i++)
    {
        ans+=sum(a[i]+n+1);
        add(a[i]+n+1,1);
    }
    
    cout<<ans<<endl;
    
    return 0;
}