【二分答案】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 位置对应的数字，’/’ 为下取整。
TT 非常想得到那只可爱的猫咪，你能帮帮他吗？

Input

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

Output

输出新数组 ans 的中位数

Sample

Input

4
1 3 2 4
3
1 10 2

Output

1
5

思路

经典的选取中位数问题，首先我们发现允许的复杂度是nlogm,故想到二分答案，二分中位数，求取小于等于该数的个数，若大于等于一半（可能有中位数有多个的情况）。求个数的方法是对原数列排序后枚举i，对[i+1,n]二分出小于等于的位置，求和若和小于一半，说明答案偏小，否则默认答案偏大。

总结

对于中位数有多个的情况的处理十分巧妙。此外此题用cin优化也会T,故用scanf。

代码

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iomanip>
using namespace std;
#define rep(i,s,t) for(int i=s;i<=t;i++)
#define mem(a,s) memset(a,s,sizeof(a))
int n;
int R;
int a[100001];
int f1(int l,int r,int x){ 
 int ans=l-1;
 while(l<=r){
  int m=(l+r)>>1;
  if(a[m]<=x){
      ans=m;
   l=m+1;
  }else r=m-1;
 }
 return ans;
}
int Ans(int x){
 int sum=0; 
 rep(i,1,n-1)//ps
  sum+=(f1(i+1,n,a[i]+x)-i);//寻找>x的第一个位置,-i后代表小于等于x的数目 
 return sum; 
}
int f(){//中位数 
 int l=0,r=R,ans=0;
 int mid=(n*(n-1)/2+1)>>1;
 while(l<=r){
  int m=(l+r)>>1;
  if(Ans(m)<mid)//<=的个数少，则偏小
   l=m+1;
  else{//多位中位数包括在内 
   ans=m;
   r=m-1;
  }
 }
 return ans; 
}
int main(){
 //freopen("1.txt","r",stdin);
 //ios::sync_with_stdio(false);
 while(~scanf("%d",&n)){
  rep(i,1,n)
   scanf("%d",&a[i]);
  sort(a+1,a+n+1);
  R=a[n]-a[1];
  printf("%d\n",f()); 
 }
 return 0;
}