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
}
}