[CQOI2009]中位数图

https://ac.nowcoder.com/acm/problem/19913

题目描述

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

输入描述

第一行为两个正整数n和b ，第二行为1~n 的排列。
对于30%的数据中，满足 n≤100；
对于60%的数据中，满足 n≤1000；
对于100%的数据中，满足 n≤100000,1≤b≤n。

输出描述

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

样例一

输入

7 4
5 7 2 4 3 1 6

输出

4

一道思维题。

输入是1~n的排列，长度为奇数的连续子序列的中位数是b，那么符合条件的子序列必然包含中位数b + 数量相等的比b大的数和比b小的数。我们不妨记大于b的数为1，小于b的数为-1。

除了必选的中位数b我们接着考虑数量相等的比b大的数和比b小的数,具体选法有三种情况：

1.都在中位数所在位置的左边选

2.都在中位数所在位置的右边选

3.同时在中位数所在位置的左右两边选

1 、2中情况为，从中位数旁边的位置开始，逐个向两边遍历，用sum记录下和值，当sum等于0说明1的个数和-1个数相等，此时符合题意。对于3，可以额外开一个内存大于a数组两倍的数组num，记录左右对称情况。在pos左边时，num[n+sum]++,在pos右边时，res += num[n-sum].

举个例子：5 7 2 4 3 1 6  --> 1 1 -1 4 -1 -1 1

const int N = 100010;
int a[N],s[N];
int num[N<<1];

void solve()
{   
    int n,b;
    cin>>n>>b;  
    int res = 1;//只选中位数自己算一个答案
    int pos;//记录中位数的位置
    for(int i=1;i<=n;i++){
        cin>>a[i];
        if(a[i]>b) a[i] = 1;
        else if(a[i]<b) a[i] = -1;
        else pos = i;
    }

    int sum = 0;
    for(int i=pos-1;i>=1;i--){
        sum+=a[i];
        num[n+sum]++;
        if(!sum) res++; //只在左边选+中位数
    }
    sum = 0;
    for(int i=pos+1;i<=n;i++){
        sum+=a[i];
        res += num[n-sum];//同时在左右选+中位数
        if(!sum) res++;//只在右边选+中位数
    }
    cout<<res;
}