题目传送门
好题。
解法:
连续子段中位数为b。
那么这段子段肯定包含b。
那么从b的位置开始可以往左边延伸也可以往右边延伸。
那么小于b的数的数量要等于大于b的数的数量。
小于b的为1,大于b的为-1。
最后整一段的和等于0那么中位数肯定是b。
用l,r。
l[i]表示从b开始向左连续一段和为i的有多少种情况。
比如l[1]等于从b这个位置开始向左连续一段和为1的有多少种情况。
r[i]就表示往右。
那么l[1]*r[-1]就是一部分的答案。
左边和为1右边和为-1加起来总和为0那么b肯定是中位数。
代码实现:
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<queue>
using namespace std;
int a[110000],l[210000],r[210000];
int main() {
int n,b,p;scanf("%d%d",&n,&b);
for(int i=1;i<=n;i++) {
int x;scanf("%d",&x);
if(x<b)
a[i]=1;
if(x>b)
a[i]=-1;
if(x==b)
p=i;
}
//数组不能为负所以都加上n。
int sum=0;l[n]=r[n]=1; //l[n]和r[n]相当于l[0]和r[0],也就是一开始不延伸和等于0的就一种情况就是只有自己。
for(int i=p-1;i>=1;i--) {
sum+=a[i];l[sum+n]++;
}
sum=0;
for(int i=p+1;i<=n;i++) {
sum+=a[i];r[sum+n]++;
}
int ans=0;
for(int i=-n;i<=n;i++)
ans+=l[n-i]*r[i+n]; //其实是l[i]*r[-i],但是每个都加了n,所以成这样了。
printf("%d\n",ans);
return 0;
}