bzoj 1303 中位数
传送门:http://www.lydsy.com/JudgeOnline/problem.php?id=1303
Description
给出1~n的一个排列,统计该排列有多少个长度为奇数的连续子序列的中位数是b。中位数是指把所有元素从小到大排列后,位于中间的数。
Input
第一行为两个正整数n和b ,第二行为1~n 的排列。
Output
输出一个整数,即中位数为b的连续子序列个数。
Sample Input
7 4
5 7 2 4 3 1 6
5 7 2 4 3 1 6
Sample Output
4
HINT
第三个样例解释:{4}, {7,2,4}, {5,7,2,4,3}和{5,7,2,4,3,1,6}
N<=100000
明显卡暴力
所以用O(n)的方法
将小于b的标记为-1,大于b的标记为1,b本身标记为0
#include<cstdio>
#include<iostream>
using namespace std;
int num[100001],sum[100001],l[200001],r[200001];
int n,b,point,ans;
int main()
{
scanf("%d%d",&n,&b);
for(int i=1; i<=n; i++)
{
scanf("%d",&num[i]);
if(num[i]>b)num[i]=1;
else if(num[i]==b)
{
num[i]=0;
point=i;
}
else num[i]=-1;
}
l[n]=1;
r[n]=1;
for(int i=point-1; i>=1; i--)
{
sum[i]=sum[i+1]+num[i];
l[sum[i]+n]++;
}
for(int i=point+1; i<=n; i++)
{
sum[i]=sum[i-1]+num[i];
r[sum[i]+n]++;
}
for(int i=0; i<=2*n-1; i++)ans+=l[i]*r[2*n-i];
printf("%d",ans);
return 0;
}