【LeetCode】33-搜索旋转排序数组

题目描述

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如，数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

搜索一个给定的目标值，如果数组中存在这个目标值，则返回它的索引，否则返回 -1 。

你可以假设数组中不存在重复的元素。

你的算法时间复杂度必须是 O(log n) 级别。

示例 1:

输入: nums = [4,5,6,7,0,1,2], target = 0
输出: 4

示例 2:

输入: nums = [4,5,6,7,0,1,2], target = 3
输出: -1

---

解题思路

一个有序数组经过旋转操作后，可以看成两部分，而这两部分又分别是有序的，我们先找到tatget落在数组的哪一半，再在那一半继续进行二分查找。

Java 实现

public int search (int[] nums, int target) {
    int n = nums.length;
    int lo = 0, hi = n - 1;
    while (lo <= hi) {
        int mid = (lo + hi) / 2;
        if (nums[mid] == target)
            return mid;

        if (nums[mid] >= nums[lo]) {
            if (target >= nums[lo] && target < nums[mid])
                hi = mid - 1;
            else
                lo = mid + 1;
        }
        else if (nums[mid] <= nums[hi]) {
            if (target > nums[mid] && target <= nums[hi])
                lo = mid + 1;
            else
                hi = mid - 1;
        }
    }
    return -1;
}

心得体会

二分查找的变种，要注意哪些地方取等哪些地方不取等。

在判断target和nums[mid]不能取等，因为它们正好相等的时候要返回mid，所以其他地方的判断不能取等。