二分法例题 LintCode62/63/159/160 搜索旋转排序数组及变体

题目描述

解题思路

本题不同于普通的升序数组，虽然数组内部是按序排列的，但是数组以某个元素作为支点进行了旋转，那么就不能直接使用二分查找，需要分情况进行讨论。
这个时候最好的办法就是在纸上画画图，将可能发生的情况都罗列出来，类似于做数学题的分类讨论，做到不重不漏。
situation1
A[mid] > A[start]
(1) A[start] < target < A[mid]

end = mid;
(2) target > A[mid]

或者target < A[start]

start = mid;

situation2
A[mid] < A[start]
(1) A[mid] < target < A[end]

start = mid;
(2) target < A[mid]

或者target > A[end]

end = mid;

Java代码实现

    public int search(int[] A, int target) {
        int n = A.length;
        if (n == 0)
            return -1;
        int start = 0, end = n - 1, mid;
        while (start + 1 < end) {
            mid = (start + end) / 2;
            if (target == A[mid])
                return mid;
            if (A[mid] >= A[start])
                if (target >= A[start] && target < A[mid])
                    end = mid;
                else
                    start = mid;
            else {
                if (target <= A[end] && target > A[mid])
                    start = mid;
                else
                    end = mid;
            }
        }
        if (target == A[start])
            return start;
        if (target == A[end])
            return end;
        return -1;
    }

变体1(LintCode63)

这是LintCode62题的变体，数组中允许出现重复的数字了，相应的算法也需要做出调整，我们在判断A[mid]和A[start]关系时需要考虑(A[mid] == A[start])的情况，由于while循环中首先要进行的就是if (target == A[mid])的判断，所以能够保证(A[mid] == A[start])时，A[start]和target一定不是相等的，那么就可以将start去掉，进一步缩小范围，也是为了去除数组中的重复元素。

Java代码实现

    public boolean search2(int[] A, int target) {
        int n = A.length;
        if (n == 0)
            return false;
        int start = 0, end = n - 1, mid;
        while (start + 1 < end) {
            mid = (start + end) / 2;
            if (target == A[mid])
                return true;
            if (A[mid] > A[start])
                if (target >= A[start] && target < A[mid])
                    end = mid;
                else
                    start = mid;
            else if (A[mid] < A[start]) {
                if (target <= A[end] && target > A[mid])
                    start = mid;
                else
                    end = mid;
            } else
                start++;
        }
        if (target == A[start] || target == A[end])
            return true;
        return false;
    }

变体二(LintCode159)

这个题目也是LintCode62题的变体，但是因为没有了target，所以其实是变简单了，只需要判断nums[mid]和nums[end]的关系即可；注意这里首先判断(nums[start] < nums[end])是否成立，如果成立则说明整个数组都是递增的，直接返回num[0]即可。

Java代码实现

    public int findMin(int[] nums) {
        int start = 0, end = nums.length - 1, mid;
        if (nums[start] < nums[end])
            return nums[start];
        while (start + 1 < end) {
            mid = (start + end) / 2;
            if (nums[mid] > nums[end])
                start = mid;
            else
                end = mid;
        }
        return Math.min(nums[start], nums[end]);
    }

变体三(LintCode160)

这个题目是LintCode159的变体，唯一的区别就在于允许数组中出现重复的元素，我们的解决思路还是像解决LintCode63一样，当(nums[mid] == nums[end])时，说明nums[end]不是最小值或者不是唯一的最小值，那么便可以将他去掉来缩小搜索范围。

Java代码实现

    public int findMin(int[] nums) {
        int start = 0, end = nums.length - 1, mid;
        if (nums[start] < nums[end])
            return nums[start];
        while (start + 1 < end) {
            mid = (start + end) / 2;
            if (nums[mid] > nums[end])
                start = mid;
            else if (nums[mid] < nums[end])
                end = mid;
            else
                end--;
        }
        return Math.min(nums[start], nums[end]);
    }