Leetcode 33. 搜索旋转排序数组 C++

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

题解

可以使用二分查找的思想，首先根据题意我们知道，数组不是严格递增或递减的，这对于二分查找是一个障碍。但我们知道数组一半是递增、一半是递增，但右部分的数严格小于左部分，我们可以利用这个性质，确定我们现在查找的部分区间属于哪一个部分，我们假设我们现在在区间i~j进行二分查找

• nums[mid]==target,查找到，直接返回
• nums[i]<=nums[mid]，即i~mid位于旋转点的左侧
  • nums[i]<=target && target<=nums[mid]，那么target就在i~mid这个区间上
  • 否则，在mid~j这个区间上
• 其他情况，即mid~j位于旋转点的右侧
  • nums[mid]<=target && target<=nums[j]，那么target就在i~mid这个区间上
  • 否则，在mid~j这个区间上

代码

int search(vector<int>& nums, int target) {
        int i=0,j=nums.size()-1;
        int mid;
        while(i<=j){
            mid = (i+j)/2;
            if(nums[mid] == target) return mid;
            else if(nums[i]<=nums[mid]){		//i~mid在旋转点左侧
                if(nums[i]<=target && nums[mid]>=target)  j=mid-1;
                else    i=mid+1;
            }else{								//mid~j在旋转点右侧
                if(nums[mid]<=target && nums[j]>=target)  i=mid+1;
                else j=mid-1;
            }
        }
        return -1;
    }

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-in-rotated-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。