Week4 C - TT 的神秘礼物

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

解题思路
将所有数排序后再相减在排序放入 ans[] 会 TL
所以我们先不做相减 也就是 ans[] 暂时还为空
先用最大值跟最小值相减 得出所有数相减会得出的可能范围
用这个范围做二分查找
用 for 遍历找ans[ j ] - ans[ i ] >= m
j>=i+m 固定 i+m 进行二分查找
返回 j
利用 i j 做出 j - i 的位置

Code

#include<iostream>
#include<stdlib.h>
#include<string>
#include<algorithm>
using namespace std;
int a[105000];int n;
int findf(int x,int y,int c,int u){//x y:范围 c: i u: m
	int f=x;int e=y;
	
	int ans=n;
	while(f<=e){
	int mid=(f+e)/2;
		if(a[mid]>=a[c]+u){//j>=i+m
			ans=mid;
			e=mid-1;
		}else{
			f=mid+1;
		}
	}
	return ans;
}

int main(){
	
	while(~scanf("%d",&n)){
		
		for(int i=0;i<n;i++){
			scanf("%d",&a[i]);
		}
		sort(a,a+n);
		int maxa=a[n-1]-a[0];
		int minia=0;
		int set=(n*(n-1)/2-1)/2;
		int ans=0;
		while(minia<maxa){
		ans=0;
			int mid=(minia+maxa)/2;
			for(int i=0;i<n;i++){
			    int t=findf(0,n-1,i,mid);//i 为 i-j 的 i ,mid 为预测 j-i 的值
				ans+=n-t;//ans 为 mid 的位置
			}
		    if(ans>=n*(n-1)/2-set){//跟中位数位置比较
		    	minia=mid+1;
			}else{
				maxa=mid;
			}
		}
		cout<<minia-1<<endl;//记录mid+1
	}
}