题意:
TT 是一位重度爱猫人士,每日沉溺于 B 站上的猫咪频道。
有一天,TT 的好友 ZJM 决定交给 TT 一个难题,如果 TT 能够解决这个难题,ZJM 就会买一只可爱猫咪送给 TT。
任务内容是,给定一个 N 个数的数组 cat[i],并用这个数组生成一个新数组 ans[i]。新数组定义为对于任意的 i, j 且 i != j,均有 ans[] = abs(cat[i] - cat[j]),1 <= i < j <= N。试求出这个新数组的中位数,中位数即为排序之后 (len+1)/2 位置对应的数字,’/’ 为下取整。
输入输出要求:
多组输入,每次输入一个 N,表示有 N 个数,之后输入一个长度为 N 的序列 cat, cat[i] <= 1e9 , 3 <= n <= 1e5
输出新数组 ans 的中位数
样例输入:
4
1 3 2 4
3
1 10 2
样例输出:
1
8
思路:
暴力做法:
枚举i, j将数列B计算出来,然后取它的中位数,时间和空间复杂度为n^2级别,考虑优化
判断是否为中位数:
要判断P是否为中位数,首先要计算P在数列中的名次。
如果P从小到大排的名次比中位数小,说明P比中位数要小
如果P从小到大排的名次比中位数大,说明P比中位数要大
如果P从小到大排的名次等于中位数,说明P就是中位数
那么问题来了,如何计算P的排名?
计算排名:
1.首先要将数列从小到大排序(去绝对值)
2.变形:计算P的名次,即计算𝑋j−𝑋i≤𝑃的二元组对数即可。 移项可得𝑋j≤𝑋i+𝑃,𝑖<𝑗
枚举下标i然后计算满足条件的下标j的个数
二分:
这里计算满足条件的下标j的个数 ,可以使用二分法找到最后一个满足条件的位置
总结:
1.第一点是判断是否为中位数的方法(名次比较)
2.这个题的关键是𝑋j≤𝑋i+𝑃的变形
3.有序都可以考虑二分
4.还有输入问题!!使用cin超时,换用了scanf,也可关同步
代码:
#include<cstdio>
#include<algorithm>
using namespace std;
int N;
int cat[100001];
int find2(int x){ //找最后一个满足Xj<=Xi+P的位置
int l=0,r=N-1,ans=-1;
while(l<=r){
int mid=(l+r)>>1;
if(cat[mid]<=x){
ans=mid; l=mid+1;
}
else r=mid-1;
}
return ans;
}
int find1(){ //二分求最小值
int l=0,r=cat[N-1]-cat[0],ans=-1;
int middle=((N-1)*N/2+1)/2; //中位数的位置
while(l<=r){
int P=(l+r)>>1; //使Xj<=Xi+P
int P_rank=0;
for(int i=0;i<N;i++){ //枚举i,找到符合条件的j
int k=cat[i]+P;
int t=find2(k);
if(t!=-1) P_rank+=t-i;
}
if(P_rank>=middle){ //名次>=中位数
ans=P; r=P-1;
}
else l=P+1;
}
return ans;
}
int main(){
while(scanf("%d",&N)!=EOF){
for(int i=0;i<N;i++)
scanf("%d",&cat[i]);
sort(cat,cat+N); //对cat数组排序
printf("%d\n",find1());
}
return 0;
}