【二分】二分算法

来源

来自中国大学MOOC郭炜老师的《程序设计与算法（二）算法基础》

经典二分

写一个函数BinarySeach， 在包含size个元素的、从小到大排序的int数组a里查找元素p,如果找到，则返回元素下标，如果找不到，则返回-1。要求复杂度O(log(n))

public static int searchRange(int[] nums, int target) {
    int L = 0; //查找区间的左端点
    int R = nums.length - 1; //查找区间的右端点
    while( L <= R) { //如果查找区间不为空就继续查找
        int mid = L+(R-L)/2; //取查找区间正中元素的下标
        if( target == nums[mid] )
            return mid;
        else if( target > nums[mid])
            L = mid + 1; //设置新的查找区间的左端点
        else
            R = mid - 1; //设置新的查找区间的右端点
    }
    return -1;
} //复杂度O(log(n))

二分小变形

找一个小于给定值的下表最大的元素

写一个函数LowerBound， 在包含size个元素的、从小到大排序的int数组a里查找比给定整数p小的，下标最大的元素。找到则返回其下标，找不到则返回-1

public static int searchRange(int[] nums, int target) {
    int L = 0; //查找区间的左端点
    int R = nums.length - 1; //查找区间的右端点
    int lastPos = -1;//到目前为止找到的最优解
    while( L <= R) { //如果查找区间不为空就继续查找
        int mid = L+(R-L)/2; //取查找区间正中元素的下标
        if(nums[mid]>= target)
            R = mid - 1;
        else {
            lastPos = mid;
            L = mid+1;
        }
    }
    return lastPos;
}

例题1：找一对数

输入n ( n<= 100,000)个整数， 找出其中的两个数，它们之和等于整数m(假定肯定有解)。 题中所有整数都能用 int 表示

解法1：双重循环n^2 = 100亿

解法2：二分，整个题目是O(nlogn)
先排序，然后枚举每个数组的元素，二分查找有没有m-a[i]

解法3：
先排序（快排O(nlogn)）
两个指针，指向数组头和尾，求a[i]+a[j]，如果大于m则移动j-1，反之移动i+1。这部分O(n)
最后，总的来说复杂度O(nlogn)

//比较简单,代码略

例题2：农夫和奶牛POJ2456

题目地址

http://poj.org/problem?id=2456

题目

农夫 John 建造了一座很长的畜栏，它包括N (2≤N≤100,000)个隔间，这些小隔间的位置为x0,…,xN-1 (0≤xi≤1,000,000,000,均为整数,各不相同).John的C (2≤C≤N)头牛每头分到一个隔间。牛都希望互相离得远点省得互相打扰。怎样才能使任意两头牛之间的最小距离尽可能的大，这个最大的最小距离是多少呢？

解法

二分查找距离D
然后尝试是否可行

import java.util.*;

public class Main {

    static int[] a = new int[100000+5];
    static int N,C;
    public static boolean isOk(int dis,int cows){
        int cnt = 1,last = a[0];
        for(int i=1;i<N;++i) {
            if (a[i] - last >= dis) {
                cnt++;
                last = a[i];
            }
            if(cnt>=cows) return true;
        }
        return false;
    }

    public static int searchRange(int n, int cows) {
        int L = 0; //查找区间的左端点
        int R = n; //查找区间的右端点
        int lastPos = 0;//到目前为止找到的最优解
        while( L <= R) { //如果查找区间不为空就继续查找
            int mid = L+(R-L)/2; //取查找区间正中元素的下标
            if(isOk(mid,cows)){
                L = mid + 1;
                lastPos = mid;
            }
            else {
                R = mid-1;
            }
        }
        return lastPos;
    }

    public static void main(String[] args){
        Scanner scanner = new Scanner(System.in);
        int i=0,j;
        N = scanner.nextInt();
        C = scanner.nextInt();
        j=N;
        while(j--!=0) {
            int temp = scanner.nextInt();
            a[i++] = temp;
        }
        Arrays.sort(a,0,N);
        System.out.println(searchRange(1000000000/C,C));
    }
}