CQOI2009中位数

                          CQOI2009中位数

给出1~n的一个排列，统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后，位于中间的数。

 

输入格式：

第一行为两个正整数n和b ，第二行为1~n 的排列。

 

输出格式：

输出一个整数，即中位数为b的连续子序列个数。

 

样例输入：

5 4
1 2 3 4 5
-----------
6 3
1 2 4 5 6 3
-----------
7 4
5 7 2 4 3 1 6

 

样例输出：

5
-----------
1
-----------
4

 

数据范围：

编号 1 2 3 4 5 6 7 8 9 10
N 10 50 100 300 1000 3600 10000 25000 55555 100000

 

时间限制：

2000

 

空间限制：

512000

 

提示：

第三个样例解释：{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}。

解析：把比中位数小的改为-1，自己为0，比它大的为1，在爆枚后用乘法原理。

#include<bits/stdc++.h>

using namespace std;

#define N 100001

#define PER(i,a,b) for(int i=a;i<=b;i++)

#define REP(i,a,b) for(int i=a;i>=b;i--)

int n;

int a[N];

int hz1[N],hz2[N];

int hy1[N],hy2[N];

int let[N];

int b,wei,ans;

int main()

{

    cin>>n>>b;

    PER(i,1,n)

    {

        cin>>a[i];

        if(a[i]<b)let[i]=-1;

        else if(a[i]==b)

        {

            let[i]=0;

            wei=i;

        }

        else let[i]=1;

    }

    int cnt=0;

    REP(i,wei,1)

    {

        cnt+=let[i];

        if(cnt>=0) hz1[cnt]++;

        else hz2[abs(cnt)]++;

    }

    cnt=0;

    PER(i,wei,n)

    {

        cnt+=let[i];

        if(cnt>=0) hy1[cnt]++;

        else hy2[abs(cnt)]++;

    }

    ans+=hy1[0]*hz1[0];

    PER(i,1,n)

    {

        ans+=hy1[i]*hz2[i];

        ans+=hy2[i]*hz1[i];

    }

    cout<<ans;

}