POJ 3579 二分答案

POJ 3579 二分答案

题目

思路

排序，二分一下答案，然后对于每一个元素看一看加上这个答案tmp之后，在数列里面排在哪里，比a[i]+tmp大的数字和a[i]的差值肯定比tmp大，统计一下就行
一开始用lower_bound统计，但是还需要考虑等于的关系，如果涉及麻烦的可以换思路用upper_bound统计，然后还有一个地方就是一半是n * (n-1) / 4 不是除以2
细节方面有很多地方有问题，要严谨

代码

#include <iostream>
#include <cstdio>
#include <set>
#include <list>
#include <vector>
#include <stack>
#include <queue>
#include <map>
#include <string>
#include <sstream>
#include <algorithm>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <cmath>
#include <fstream>
#include <iomanip>
//#include <unordered_map>
using namespace std;
#define dbg(x) cerr << #x " = " << x << endl;
typedef pair<int, int> P;
typedef long long ll;
#define FIN freopen("in.txt", "r", stdin);

ll n;
const int MAXN = 1e5+ 5;
ll a[MAXN];

ll judge(ll mid)
{
	ll cnt = 0;
	for(int i = 0; i < n; i++)
	{
		ll tmp = a[i] + mid;
		int pos = upper_bound(a, a+n, tmp) - a;
		cnt += n - pos;
	}
	//dbg(cnt);
	return cnt > n*(n - 1) / 4;
	
}
int main()
{
	//FIN;
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    while(~scanf("%d", &n))
    {
    	ll maxx = -1;
    	for(int i = 0; i < n; i++)
		{
			scanf("%lld", &a[i]);
		}
		sort(a, a+n);
		ll l = 0, r = a[n-1] - a[0];
		while(l < r)
		{
			ll mid = (l + r) >> 1;
			//dbg(mid);
			if(judge(mid))
			{
				l = mid + 1;
			}
			else
			{
				r = mid;
			}
		} 
		printf("%lld\n", l);
	} 
}