[CQOI2009]中位数图

[CQOI2009]中位数图
https://ac.nowcoder.com/acm/contest/71881/G

思路：

• 题目要求的是长度为奇数的连续子序列并且中位数是b，就可以想到是以b这个位置进行向两边进行扩展查找，但是因为n的数据范围在1e5，那么暴力的方法就不可以了。
• 再往下想的话，可以发现它只是要求b是当前序列的中位数，那么就可以将原数组进行更改，大于b的值为1，小于b的值为-1，那么这样的话，在b之前以及b之后都可以单独进行++看sum的值是不是等于0就行了。
• 但是如果这个序列是在b的左右两边取的话这样就不满足了，那么就需要进行改进。
• 令左边的和为sum1，右边的和为sum2
• 假设sum1为-2，那么如果我右边的sum2值为2的话，是不是就刚好是两边的和为0，那么根据这个思路我们就可以再添加一个数组用来存储左边数据和的个数，然后在计算右边的时候进行比较就可以了
  代码：

#include<bits/stdc++.h>
using namespace std;
#define int long long

int a[100001],num[200001];
//a数组用来存储当前信息，num数组用来存储和的信息
void solve(){
	int n,b;
	cin>>n>>b;
	int mid;
	for(int i=0;i<n;i++){
		cin>>a[i];
		if(a[i]==b){
			mid=i;//找出来中间值的位置
		}
		else
		if(a[i]<b){
			a[i]=-1;//小于b就赋值为-1
		}
		else
		a[i]=1;//大于b就赋值为1
	}
	int sum=0;//sum是用来求和的
	int ans=1;//ans是中位数为b的连续子序列个数。
    //赋值为1是因为如果只有b这个元素也符合题意
	for(int i=mid-1;i>=0;i--){//在b的左边进行查找
		sum+=a[i];
		num[n+sum]++;
        //因为数组没有负数，所以让sum加上一个n
        //其实也就是num[sum]++
        //num数组就是用来存储当前sum值有多少个，方便在下面的时候进行比较
		if(sum==0){//中位数也就是b，加上左边的数满足的情况
			ans++;//个数++
		}
	}
	sum=0;//重新赋值为0，因为要开始计算右边的了
	for(int i=mid+1;i<n;i++){
		sum+=a[i];
		ans+=num[n-sum];
        //有一个n也是跟上面一样，因为数组没有负数，所以让-sum加上一个n
        //为啥是-sum呢，因为假设左边的值是-2的话，那么我右边的值应该是2才能够满足，那么不就是-sum么
		if(sum==0){中位数也就是b，加上右边的数满足的情况
			ans++;
		}
	}
	cout<<ans;
}
signed main(){
	int T=1;
	//cin>>T;
	while(T--){
		solve();
	}
	return 0;
}