描述
给出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(n2)。
#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;
}