题意
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 非常想得到那只可爱的猫咪,你能帮帮他吗?
输入输出格式:
inout: 多组输入,每次输入一个 N,表示有 N 个数,之后输入一个长度为 N 的序列 cat, cat[i] <= 1e9 , 3 <= n <= 1e5
output:输出新数组 ans 的中位数
输入输入样例:
inout:
4
1 3 2 4
3
1 10 2
output:
1
8
思路
- 此题既然是求中位数,那么如果暴力将所有的绝对值求出来,然后进行求中位数,但是时间和空间复杂度均为O(n2),如果数据量很大,那就是让人难以接受的;
- 我们可以发现,如果我们给出一个数P,如果这个数的名次大于中位数的名次,那么它比中位数大,如果小于,那么就是比中位数小,如果等于的话,那么P就是中位数;这个过程是单调的,那么便可以二分P;
- 那么P如何得到呢,我们既然拿P和中位数相比,那么P肯定也是绝对值的产物;P=Xj-Xi,既然是绝对值,那么我们就可以使得Xj-Xi<=P(Xi,Xj的对数就是P的名次)对于给定的Xi,求满足条件的Xj的个数,然后不同的Xi累计相加即就是名次;
- 那么只需要看Xj<=P+Xi(条件)的个数,则先进行一个排序,只需要知道,第一次Xj<=P+Xi的位置,和第一次Xj>P+Xi的位置即可,相减就是一次给定的Xi得到的数对数目,循环n次就可以得到P的名次;
- 此时P每次都是进行二分,取mid,如果P的名次大于等于中位数的名次,那么 ans=mid,然后right=mid-1;如果小于的话,left=mid+1;一直二分到left>right停止二分。最后ans就是结果
总结
- 此题通过对答案的一个分析,发现可以进行二分答案范围,一直二分便可以得到想要的答案;
- 此题的一个疑惑是为撒(如果P的名次大于等于中位数的名次,那么 ans=mid,)难道不能小于等于吗,此时解释如果得到中位数是11122222,中位数是2,此时P=(left+right)>>2=2,名次大于等于中位数的名次 ,那么ans=2,然后right=1;然后P=1;然后1的名次小于中位数名次,此时left=2;退出二分,此时ans无误;
- 如果小于等于ans=mid的话;那么ans=1了,出现错误;
代码
#include<cstdio>
#include<algorithm>
using namespace std;
int last_location(int left,int right,int *q,int x)
{
int ans = -1;
while (left <= right)
{
int mid = (left + right) >> 1;
if (q[mid] <= x)
{
ans = mid;
left = mid + 1;
}
else
right = mid - 1;
}
return ans;
}
int p[100001];
int main()
{
int N;//
while (~scanf_s("%d",&N))
{
for (int i = 0; i < N; i++)
scanf_s("%d", &p[i]);
sort(p, p + N);
int rank1 = (N * (N - 1) / 2 + 1) / 2 ;
int ans = -1;
int left = 0;
int right = p[N - 1] - p[0 ];
while (left <= right)
{
int rank = 0;
int mid = (left + right) >> 1;
for (int i = 0; i < N; i++)
{
int temp = last_location(0, N - 1, p, p[i] + mid);
if(temp!=-1)
rank = rank +temp-i;
}
if (rank < rank1)
left=mid+1;
else {
right = mid - 1;
ans = mid;
}
}
printf("%d\n", ans);
}
return 0;
}