题意
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 位置对应的数字,’/’ 为下取整。
TT 非常想得到那只可爱的猫咪,你能帮帮他吗?
Input
多组输入,每次输入一个 N,表示有 N 个数,之后输入一个长度为 N 的序列 cat, cat[i] <= 1e9 , 3 <= n <= 1e5
Output
输出新数组 ans 的中位数
思路
这是一道二分答案的题,ans[] = abs(cat[i] - cat[j]),我们要求ans数组的中位数,首先我们将cat数组由小到大排序之后可以将绝对值去除,我们可以得到
cat[j] - cat[i] = p,移项得到cat[j] = cat[i] + p,p最大值为cat[n-1] - cat[0]
由中位数的性质我们可以知道,如果p要是中位数的话,那么小于等于p的值由 (len+1)/2 个,所以cat[j] <= cat[i] + p , (j > i),cat[j]也有 (len+1)/2,在此题中为 (n * (n - 1)/2 - 1)/2个。
由于p范围已知[0, cat[n-1] - cat[0] ],所以我们对p进行二分,对于二分的这个p,对cat中每一个i值,我们再次进行二分,来寻找满足cat[j] <= cat[i] + p的cat[j]的个数,所有的个数加起来就是p再ans数组中的位次,与上面的中位数位次比较,继续对p进行二分,直到结束。
总结
第一次接触二分答案的题,感觉比较新颖,主要是判断如何对答案进行二分。
代码
#include<iostream>
#include<algorithm>
using namespace std;
int cat[100005];
int binary(int *cat,int i,int p,int n)
{
int ans=-1;
int l=i+1,r=n-1;
while(l<=r)
{
int mid=(l+r)>>1;
if(cat[mid]<=cat[i]+p)
{
l=mid+1;
ans=mid;
}
else
r=mid-1;
}
return ans-i;
}
int main()
{
int n;
while(scanf("%d",&n)!=EOF)
{
for(int i=0;i<n;i++)
scanf("%d",&cat[i]);
sort(cat,cat+n);
int l=0,r=cat[n-1]-cat[0];
int midr=(n*(n-1)/2+1)/2;
while(l<=r)
{
int mid=(l+r)>>1;
int rank=0;
for(int i=0;i<n;i++)
{
int temp=binary(cat,i,mid,n);
if(temp>=0)
rank+=temp;
}
if(rank>=midr)
r=mid-1;
else
l=mid+1;
}
printf("%d\n",r+1);
}
}