搜索旋转排序数组

搜索旋转排序数组

来源Leetcode

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

( 例如，数组 [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

题解

看到O(lgn)，就想到查找算法中的二分查找。这是一个二分查找的变种，二分查找的适用范围是有序序列或者局部有序序列。

思路是先判断左半部分有序，还是右半部分有序

如果左半部分有序，并且target又位于左半部分区间里，则r = mid - 1在左边区间查找，否则 l = mid + 1

如果右半部分有序，并且target又位于右半部分区间里，则l = mid + 1 在右边区间查找，否则 r = mid - 1

代码

class Solution {
public:
    int search(vector<int>& nums, int target) {
        if(nums.size()==0) return -1;
        int l = 0,r = nums.size()-1;
        while(l<=r)
        {
            int mid = (l+r)/2;
            // printf("%d %d %d\n",l,r,mid);
            if (nums[mid] == target) return mid;
            if(nums[l]<=nums[mid]){  //左半边有序,这个等号搞了我半个小时
                if(target<nums[mid] && nums[l]<=target) r = mid - 1;
                else l = mid + 1;
            }else{
                if(target>nums[mid] && nums[r]>=target) l = mid + 1;
                else r = mid - 1;
            }
        }
        return -1;
    }
};