线性表-Search in Rotated Sorted Array

描述

Suppose a sorted array is rotated at some pivot unknown to you beforehand.
(i.e., 0 1 2 4 5 6 7 might become 4 5 6 7 0 1 2).
You are given a target value to search. If found in the array return its index, otherwise return -1.
You may assume no duplicate exists in the array.

分析

使用二分查找,注意边界条件

代码实现 c++
class solution{
pubic: 
    int search(const vector& nums,int target){
        int first = 0,last = nums.size();
        while(first != last){
            const int mid = first + (last - first) / 2;
            if (nums[mid] == target)
                return mid;
            if (nums[first] <= nums[mid]){
                // 注意此处有等于号,表明在闭区间内
                if(nums[first] <= target && target < nums[mid])
                    last = mid;
                 else
                    first = mid + 1;
             } else {
                    if(nums[mid] < target && target <= nums[last - 1])
                        first = mid + 1;
                    else
                        last = mid;
              }
          }
          return -1;
    }
};
代码实现 python
class Solution(object):
    def search(self, nums, target):
       if not nums:
            return -1
            
        start, end = 0, len(nums) - 1
        while start + 1 < end:
            mid = int(start + (end - start) / 2)
            if nums[mid] == target:
                return mid
            elif nums[mid] > nums[start]:
                if target >= nums[start] and target <= nums[mid]:
                    end = mid
                else:
                    start = mid
            else:
                if target >= nums[mid] and target <= nums[end]:
                    start = mid
                else:
                    end = mid
                    
        if nums[start] == target:
            return start
        if nums[end] == target:
            return end
        return -1

你可能感兴趣的:(线性表-Search in Rotated Sorted Array)