POJ 3579 Median

POJ 3579 Median

题意：给你n个数，对每对数进行计算∣Xi - Xj∣(1 ≤ i ＜ j ≤ n)，总共有m对，求第（m/2）小的数是多少

思路：由于有绝对值，所以对给定的n个数排序不影响结果。然后设l=0,r=a[n-1]-a[0],对答案进行二分，看有多少小于mid的数，在查找小于mid的数的个数时也要进行二分，不然还是会超时。相当于两重二分，写第二重时一直写不出来，就百度了C++upper_bound的源码修改了下，写进去了，多机智

AC代码：

import java.util.Arrays;
import java.util.Scanner;

public class Main
{
	static Scanner scan=new Scanner(System.in);
	
	private static int upper_bound(int[] a,int l,int r,int key)
	{
		int len=r-l,half,mid;
		while(len>0)
		{
			half=len>>1;
			mid=l+half;
			if(a[mid]>key) len=half;
			else {l=mid+1;len=len-half-1;}
		}
		return l;
	}
	
	public static void main(String[] args)
	{
		while(scan.hasNext())
		{
			int n=scan.nextInt(),m=(n*(n-1)/2+1)>>1;
			int a[]=new int[n];
			for(int i=0;i<n;i++) a[i]=scan.nextInt();
			Arrays.sort(a);
			int count1,count2,p;
			int l=0,r=a[n-1]-a[0],mid=(l+r)>>1;//mid表示要找的最小数是多少
			for(;l<=r;mid=(l+r)>>1)
			{
				count1=count2=0;
				for(int i=0;i<n-1;i++)
				{
					p=upper_bound(a,i+1,n,a[i]+mid);//找~[i+1,n-1]区间里，第一个严格大于a[i]+mid的数
					count2+=n-p;//count2记录所有符合以上规律的数
					for(int j=p-1;j>i;j--)
						if(a[j]==a[i]+mid) p--;//等号的去掉
						else break;
					count1+=p-i-1;//count1记录所有严格小于a[i]+mid的数
				}
				//if(count1<m && count2<m) break;//如果满足这种情况，说明答案已经出来了，当然，去掉也无所谓
				if(count1<m) l=mid+1;//如果小于a[i]+mid的数的个数小于m,表示mid取小了
				else r=mid-1;//二分答案，不断逼近上下界
			}
			System.out.println(mid);
		}
	}
}