BZOJ 1303
【题目链接】
http://www.lydsy.com/JudgeOnline/problem.php?id=1303
【解题报告】
思路比较巧妙,重新定义数组,比b大的数设为-1,比b小的数设为1。
为什么可以这么想呢?对于中位数而言,我们只关心这个连续序列里有多少个元素比他大,而不关心比他大多少,即比他大的数无论怎么排列都没有关系。
也即,数列里的数只有三个性质:大于b,小于b,等于b。
那么我们使用-1,0,1即可表示数列里的数的性质。
为什么要维护这个性质呢?
因为我们需要在O(N)时间内求得以b为中位数的区间有多少个。
也即,对一个区间,我们需要O(1)查询b是不是中位数。
这里的思路最关键。我们设pos为b所在的位置,那么在区间(l,r)内,
我们知道(pos+1,r)的sum。
sum=0,大于b和等于b的数一样多
sum>0,大于b的数比小于b的数多sum个
sum<0,同理
那么如果sum(l,pos-1)+sum(pos+1,r)=0,是不是说,在(l,r)内大于b和小于b的数是一样多的?
所以我们只需要跑一遍前缀和即可。
还有一个细节,需要对数组下标进行平移。读者可以在编写代码的过程中自行分析和设计。
【参考代码】
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int N,b;
int hashL[200000+50],hashR[200000+50];
int sum[100000+50],a[100000+50];
int main()
{
while( ~