中位数图（前缀和）

题目描述

链接：https://ac.nowcoder.com/acm/problem/19913
来源：牛客网

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

主要思路：

这道题刚看见的时候我首先想到的是双指针算法，但是想了想，发现我不会写。后来看了一下别人的代码，发现了一个很简单但是我没想到的思路。
就是首先找到b在排列中的位置，然后分别向左向右的扩散计算前缀和。左右两边的融合就利用总和为0这一特性进行计算，由于数组的下标不能为负数，记录时全部加上n

#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
const int N=1e5+10;
int a[N],num[N];
int main()
{
    int n,m;
    cin>>n>>m;
    int pos;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        if(x>m) a[i]=1;
        else if(x<m) a[i]=-1;
        else pos=i;
    }
    // 先记录左边的数量
    int sum=0;
    int ans=1;
    for(int i=pos-1;i;i--)
    {
        sum+=a[i];
        if(sum==0) ans+=1;
        num[n+sum]+=1;
    }
    sum=0;
    // 记录右边的数量与左右结合的数量
    for(int i=pos+1;i<=n;i++)
    {
        sum+=a[i];
        if(sum==0) ans+=1;
        ans+=num[n-sum];
    }
    cout<<ans<<endl;
    return 0;
}