中位数图-前缀和应用

链接：登录—专业IT笔试面试备考平台_牛客网
来源：牛客网
 

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

7 4
5 7 2 4 3 1 6

输出

4

备注:

对于30%的数据中，满足 n≤100；

对于60%的数据中，满足 n≤1000；

对于100%的数据中，满足 n≤100000,1≤b≤n。

分析：

        直接枚举肯定超时，开始想了很长时间一点思路没有，后来发现可以用前缀和来解决，与数组本身除了b以外其他的值无关，可以将数组中小于b的值化为-1，大于b的值化为1，构建前缀和数组，枚举区间左右端点，如果区间和等于b则表明区间中位数为b。

代码：

#include <iostream>
#include <cstdio>

using namespace std;

const int N=1e5+10;

int a[N];

int main()
{
    int n,b;
    
    scanf("%d %d",&n,&b);
    
    for(int i=1;i<=n;i++)
    {
        int x;
        scanf("%d",&x);
        if(x>b) a[i]=1;
        else if(x<b) a[i]=-1;
        else a[i]=x;
        a[i]+=a[i-1];
    }
    
    int ans=0;
    for(int l=1;l<=n;l++)
    {
        for(int r=l;r<=n;r++)
        {
            if(a[r]-a[l-1]==b) ans++;
        }
    }
    
    printf("%d",ans);
    
    return 0;
}