程序设计思维 week4 作业C-TT 的神秘礼物

题目

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 位置对应的数字，’/’ 为下取整。

Input

多组输入，每次输入一个 N，表示有 N 个数，之后输入一个长度为 N 的序列 cat， cat[i] <= 1e9 , 3 <= n <= 1e5。

Output

输出新数组 ans 的中位数。

Sample Input

4
1 3 2 4
3
1 10 2

Sample Output

1
8

思路

题意即，给定一个数组cat[]，任选其中两个数作差，并求绝对值，将所有的差的绝对值组成一个新的数组ans[]，求ans[]的中位数。因为|a-b|=|b-a|，所以可以将题目转化为将cat[]升序排序，任取j>i，求cat[j]-cat[i]，组成ans[]，求其中位数。
暴力做法为将数组ans[]算出来，并求其中位数，复杂度为O(n^2)，无法接受。
由于已知cat[]数组的元素个数n，故可以得出ans[]数组的元素个数为n*(n-1)/2，即可求中位数所在位置的下标pos=((n-1)*n/2+1)/2。这样，求解过程可以用两个二分实现。
第一个二分过程，找到中位数。设左边界l为1，右边界r为ans中最大的值即cat[n-1]-cat[0]。当l<r时，循环，mid=(l+r)/2，若值为mid的点在ans[]的排名小于等于中位点，则l=mid+1，rslt=mid；否则r=mid-1。在循环过程中，ans不一定存在于ans[]中，但是，通过不断的移动左右边界来逼近，在循环结束时一定可以将rslt定为一个存在于ans[]的数。例如，ans[]中的某一段序列为a、b、c，其中b为中位数，则对于属于(0,a]的数，其排名小于pos；对于属于(a,b]的数，其排名等于pos；对于属于(a,+inf)的数，其排名大于pos。故在<=时更新rslt。
第二个二分过程，找到名次。显然，在上述过程中，由于不能直接求出ans[]，需要设计一个查找x在ans[]中名次的函数，即计算cat[j]-cat[i]<x的二元组对数，移项可得cat[j]<cat[i]+x。枚举下标i，计算符合条件的下标j的个数。

代码

#include <cstdio>
#include <algorithm>
using namespace std;

int n;
int cat[100005];

int Rank(int x){//查找x在ans[]的名次
    int rank=0;
    for(int i=0;i<n;i++){
        int l=i+1,r=n-1,ans=n;
        while(l<=r){
            int j=(l+r)/2;
            if(cat[j]<cat[i]+x){
                l=j+1;
            }
            else{
                r=j-1;
                ans=j;
            }
        }
        rank+=(ans-i-1);//去除cat[j]-cat[i]<=0的部分
    }
    return rank+1;
}

int find(){
    int l=1,r=cat[n-1]-cat[0],pos=((n-1)*n/2+1)/2,ans=0;
    while(l<=r){
        int mid=(l+r)/2;
        if(Rank(mid)<=pos){
            l=mid+1;
            ans=mid;
        }
        else{
            r=mid-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);
        int ans=find();
        printf("%d\n",ans);
    }
    return 0;
}

题目链接