算法实验T11——POJ3579 Median

题目链接

题目大意

给出一个数列，然后计算数列里各个数之间的差值的绝对值，形成一个新数列，求新数列的中位数。

思路

        先将数组排序，数列里的差值最大为，最小为相邻两数之间差值的最小的一个。于是我们可以在这之间进行二分答案：取L 为最小差值，R为最大差值，每次对mid检查，如果对于当前的mid有半数以上的差值比mid大，则L++；否则R++。

        检查过程使用了lower_bound函数：这个函数返回给定区间中第一个大于等于给定值的地址。

        那么对于0~n-1的每一个i，lower_bound(a + i + 1, a + n, a[i] + middle) - a就是与a[i]差值小于middle的数的总数，再用n去减，得到的就是与a[i]差值大于等于middle的数的总数，全部加起来就是差值绝对值大于等于middle的总数。

AC代码

#include<iostream>
#include<algorithm>
#include<cstdio>
using std::sort;
using std::lower_bound;
typedef long long ll;
int check(ll a[], ll middle, ll m);
ll m;
int main(){
	ll n;
	ll a[100005];
	while((scanf("%lld", &n) != EOF)){
		m = n * (n - 1) / 4;
		for(ll i = 0; i < n; i++){
			scanf("%lld", &a[i]);
		}
		sort(a, a + n);
		ll minn, maxn;
		minn = a[1] - a[0];
		maxn = a[n - 1] - a[0];
		for(ll i = 1 ; i < n - 1; i++){
			if(a[i + 1] - a[i] < minn){
				minn = a[i + 1] - a[i] ;
			}
		}
		ll l = minn, r = maxn, mid;
		while(l <= r){
			mid = (l + r) >> 1;
			if(check(a, mid, n)){
				l = mid + 1;
			}
			else r = mid - 1;
		}
		printf("%lld\n", r);
	}
	return 0;
}
int check(ll a[], ll middle, ll n){
	ll num = 0;
	for(ll i = 0; i < n; i++){
		num += a + n - lower_bound(a + i + 1, a + n, a[i] + middle);
	}
	return num >= m + 1;
}