中位数——数论
题目来源
题目描述
给出1~n的一个排列,统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后,位于中间的数。
输入输出格式
输入格式:
第一行为两个正整数n和b,第二行为1~n的排列。
输出格式:
输出一个整数,即中位数为b的连续子序列个数。
输入输出样例
输入样例#1:
7 4
5 7 2 4 3 1 6
输出样例#1:
4
数据规模
对于30%的数据中,满足n≤100;
对于60%的数据中,满足n≤1000;
对于100%的数据中,满足n≤100000,1≤b≤n。
解题思路
- 题目是要求以 b 为中位数连续的子序列的个数。一个序列的中位数是将这个序列排序以后位于中间位置的数据,那么这个序列中大于中位数的数据个数一定与小于中位数的数据个数相等。
- 找到数据 b 在序列中的位置(下文以 id 代替)。利用前缀和统计位于 id 左右两侧大于 b 的数据个数与小于 b 的数据个数之差。那么 -1 就代表小于 b 的数据个数比大于 b 的数据个数多一; 同理 +1 就代表大于 b 的数据个数比小于 b 的数据个数多一,以此类推
- 在 id 的左侧和右侧分别找到两个统计结果之和为 0 的数据,以他们为起点和终点的连续子序列的中位数既是 b
- 由于 id 处对应的值与 b 相等,所以 id 对应的统计结果为 0
- 统计时有一个小技巧 :
- 首先统计位于 id 左侧的数据,在统计时还要计算出对应值出现的个数
- 在统计右侧的数据时可以边统计边计算:当利用前缀和求出该位置的数据之后,将结果累加上左侧的与此统计结果相反的个数
- 提示:由于数组下标不存在负数,可以使用 STL 提供的 map 映射 或者将下标累加一个较大的常数
源代码
#include <cstdio>
#include <iostream>
#include <map>
using namespace std;
int n,b;
int data[100005];
int value[100005];
map <int, int> Left;// 使用 map 映射统计左侧对应值出现的个数
int ans = 0; // 结果累加器
int main() {
freopen("in.txt", "r", stdin);
cin >> n >> b;
int id;
for (int i = 1; i <= n; i++) {
scanf("%d",&data[i]);
if (data[i] == b) // 找到 b 的位置
id = i;
}
value[id] = 0; // id 处对应的统计结果为0
for (int i = id; i > 0; i--) { // 利用前缀和统计左侧个数
if (data[i] > b)
value[i] = value[i + 1] + 1;
else if (data[i] < b)
value[i] = value[i + 1] - 1;
Left[value[i]]++; // 统计此个数的出现次数
}
for (int i = id; i <= n; i++) { // 利用前缀和统计右侧个数
if (data[i] > b)
value[i] = value[i - 1] + 1;
else if (data[i] < b)
value[i] = value[i - 1] - 1;
ans += Left[-value[i]]; // 结果累加
}
cout << ans;
return 0;
}