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

Sample Output

1
8

思路分析

此题最容易想到的是暴力算法， 即分别枚举x_i和x_j，得到新数组ans[]后排序找到中位数，这样复杂度是O(n^2)，对于1e5的数据来说复杂度是不能接受的。

课上学长讲到二分答案的方法，可以将复杂度降到n(logn)^2，感觉有一丢丢绕，下面详细写一写。

1.二分答案：因为直接枚举求中位数复杂度太高，可以用验证的方法，即给一个数P，判断其是否是中位数。中位数即名次排在一个有序数列中间的数，因此我们可以求出P的名次，如果小于中位数的名次，则P比中位数小，如果P的名次比中位数大，则P大于中位数，如果P的名次与中位数相同，则P就是中位数。我们可以看到P的取值是单调的，因此可以二分P来寻找与中位数名次相同的P。二分时，先将cat[]数组排序，可以去绝对值，P=x_j-x_i（i<j），这样，P最小取0，最大取cat[n-1]-cat[0]，然后一直二分直到P的名次与中位数名次相同。

2.求P的名次：在单调数列B中，P=x_j-x_i，P的名次就为小于等于P的数的个数，即求满足x_j-x_i<=P的二元组x_j、x_i的个数。变形一下：x_j<=x_i+P，因此，我们可以枚举每个x_i，找到满足上述不等式的x_j的个数即P的名次。

3.再次二分：寻找x_j的过程又可以二分数组cat[]得到，用满足条件的mid来更新pos，最后得到区间[i,pos]的长度就是满足条件的x_j的个数，也是P的名次。

4.求中位数的名次：数列B中有n(n-1)/2个数，中位数的名次为(n(n-1)/2+1)/2，位运算表示为((n*(n-1)>>1)+1)>>1。

代码

#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;

const int maxn=1e5+5;
int cat[maxn]; 
int N;
int find(int i,int B)
{
 int l=i+1,r=N-1，pos=-1;
 if(cat[l]>B) return 0;
 while(l<=r){
  int mid=(l+r)>>1;
  if(cat[mid]<=B)
  {//求最后一个<=B的位置 
   pos=mid;
   l=mid+1;
  }
  else
   r=mid-1;
 } 
 return pos-i;//区间[i,pos]的长度就是满足条件的x_j的个数 
} 
int find_p(int index)
{//二分答案 
 int l=0,r=cat[N-1]-cat[0];
 int sum=0;
 while(l<=r){
  int mid=(l+r)>>1;
  sum=0;
  for(int i=0;i<N-1;i++)
  {
   sum+=find(i,cat[i]+mid);
  }    
  if(sum<index)
   l=mid+1;
  else
   r=mid-1;
 } 
 return l;
}
int main()
{
 
 while(~scanf("%d",&N)){
  for(int i=0;i<N;i++)
  {
   scanf("%d",&cat[i]);
  }
  sort(cat,cat+N);
  int mid_index=((N*(N-1)>>1)+1)>>1;
  printf("%d\n",find_p(mid_index));
 }
 return 0;
}