牛客——中位数图（连续子数组和二维前缀和）

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

题目描述

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

输入描述:

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

输出描述:

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

#include<bits/stdc++.h>
using namespace std;
int a[100005],b[100005];
int main()
{
	int n,c;cin>>n>>c;
	int po,sum=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		if(a[i]==c) po=i;
		if(a[i]<c) b[i]++;
		if(a[i]>c) b[i]--;
		b[i]+=b[i-1];
	}
//	for(int i=1;i<=n;i++) b[i]+=b[i-1];
	for(int i=po;i<=n;i++){
		for(int j=0;j<po;j++){
			if(b[i]-b[j]==0) sum++;
		}
	}
	cout<<sum;
}