(C/C++)统计1~n的排列有多少个长度为奇数的连续子序列的中位数是b

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

输入格式
第一行为两个正整数n和b ，第二行为1~n 的排列。
输出格式
输出一个整数，即中位数为b的连续子序列个数。
样例1
样例输入1
5 4
1 2 3 4 5
样例输出1
2
样例2
样例输入2
6 3
1 2 4 5 6 3
样例输出2
1
样例3
样例输入3
7 4
5 7 2 4 3 1 6
样例输出3
4
限制
各个测试点1s
提示
第三个样例解释：{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}。
对于40%的数据，n<=300;
对于80%的数据，n<=25000;
对于全部的数据，n<=100000。

来源
NOI2009重庆市代表队选拔赛第一题

思路

一开始做的时候只想到了用两个数组记录下第i个数之前有min[i]个小于中位数的数，有max[i]个大于中位数的数，这样可以得出i到j下标之间，大于中位数的个数和小于中位数的个数。起点由0遍历至中位数所在处下标pos，终点从pos遍历至末尾，复杂度的话就是O(n²)。

#include <iostream>
#include <string>
using namespace std;
int nums[100000];
int min[100000] = { 0 };
int max[100000] = { 0 };
int main(void)
{
	int count = 0;
	int mid = 0;
	int pos = 0;
	int res = 0;
	cin >> count >> mid;
	for (int i = 0; i < count; i++)
	{
		cin >> nums[i];
		if (nums[i] > mid)
		{
			max[i] = i == 0 ? 1 : max[i - 1] + 1;
			min[i] = i == 0 ? 0:min[i - 1];
		}
		else if (nums[i] < mid)
		{
			min[i] = i == 0 ? 1 : min[i - 1] + 1;
			max[i] = i == 0 ? 0 : max[i - 1];
		}
		else
		{
			pos = i;
			max[i] = max[i - 1];
			min[i] = min[i - 1];
		}
	}
	for (int i = 0; i <= pos; i++)
	{
		for (int j = pos; j < count; j++)
		{
			int left = 0;
			int right = 0;
			if (nums[i] > mid)
			{
				left = min[j] - min[i];
				right = max[j] - max[i] + 1;
			}
			else if (nums[i] < mid)
			{
				left = min[j] - min[i] + 1;
				right = max[j] - max[i];
			}
			else
			{
				left = min[j] - min[i];
				right = max[j] - max[i];

			}
			if (left == right)
			{
				//cout << i << " " << j << endl;
				res += 1;
			}
		}
	}
	cout << res;
	system("pause");
	return 0;
}

但这种题肯定是有线性时间解的，对于中位数，可以将小于中位数的数视作-1，反之为1，转换为有多少个区间和为0的问题，并且该区间包括中位数的位置。
因为C++的数组下标不支持负数，所以将所有数+count，即left_num[i+count]，（-count<=i<=count）即中位数左侧和，以中位数为结尾的和为i的区间的个数。
left_num[0]=right_num[0]=1，因为中位数自身也算一个符合要求的区间。

#include <iostream>
#include <string>
using namespace std;
int nums[100000];
int left_num[200000];
int right_num[200000];
int main(void)
{
	int count = 0;
	int mid = 0;
	int pos = 0;	//中位数的下标
	int res = 0;
	cin >> count >> mid;
	for (int i = 0; i < count; i++)
	{
		int temp;
		cin >> temp;
		if (temp > mid)
			nums[i] = 1;
		else if (temp < mid)
			nums[i] = -1;
		else
		{
			nums[i] = 0;
			pos = i;
		}
	}
	int sum = 0;
	left_num[count] = right_num[count] = 1;
	for (int i = pos - 1; i >= 0; i--) 
	{
		sum += nums[i];
		left_num[count + sum] += 1;
	}
	sum = 0;
	for (int i = pos + 1; i < count; i++)
	{
		sum += nums[i];
		right_num[count + sum] += 1;
	}
	for (int i = - count; i <= count; i++)
	{
		res += left_num[count + i] * right_num[count - i];
	}
	cout << res << endl;
	system("pause");
	return 0;
}