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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章