题意
- 给定1~n的排列和中位数b,构造子列使得中位数是b
思路
- 比b大的记为1,小的记为-1,对b左右做后缀和和前缀和,统计每种答案个数,将前后和中互为相反数的相乘
AC代码
#include<bits/stdc++.h>
using namespace std;
#define base 100000
int a[101010];
int l[101010],r[101010],ctl[303030],ctr[303030];//记录左半边和右半边出现频率,反着记方便相乘
int main(){
int n,b,pos,ans=0;
scanf("%d%d",&n,&b);
for(int i=0;i<n;i++){
int tp;
scanf("%d",&tp);
if(tp>b) a[i]=1;
else if(tp<b) a[i]=-1;
else {a[i]=0;pos=i;}
// printf("%d ",a[i]);
}
for(int i=pos;i>=0;i--){
l[i]=l[i+1]+a[i];
ctl[base+l[i]]++;
}
for(int i=pos;i<n;i++){
r[i]=r[i-1]+a[i];
ctr[base-r[i]]++;
}
for(int i=0;i<303030;i++){
ans+=ctr[i]*ctl[i];
}
printf("%d",ans);
return 0;
}
补充:中位数和带权中位数
-
对于数轴上若干点,将它们移动到同一个点,欲使移动距离最小,应将他们移到中位数所在的位置
-
证明如下:对于移动到每一个位置,可知总距离为:
-
不妨认为已经将全部点移动到权值中位数位置,此时,左右权值相同设为
,无论向左或向右移动,均有
的权值要多移动而
的权值要少移动,故该点为最优解
-
tips:对于一下其它形式的问题,可通过将所有点先移到最左,变形上述式子,求出每一个位置的总距离,最后
遍历一遍序列即可得到最小值