LeetCode第33题——search in rotated array的O(logn)解法

问题描述
Suppose an array sorted in ascending order 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.

Your algorithm’s runtime complexity must be in the order of O(log n).

Example 1:

Input: nums = [4,5,6,7,0,1,2], target = 0
Output: 4
Example 2:

Input: nums = [4,5,6,7,0,1,2], target = 3
Output: -1
解法思路
时间复杂度O(logn)很容易联想到二分查找。但此序列不是严格的递增或者递减序列,不能直接用。但可以利用二分查找,找到序列的旋转点,将序列分为两个单调递增的序列,再分别进行二分查找,总时间复杂度为O(logn)。此题注意二分查找的循环操作!
C++代码

class Solution {
public:
    int search(vector<int>& nums, int target) {
        //二分查找思想
        int n=nums.size();
        if(n==0) return -1;
        if(n==1)
        {
            if(nums[0]==target) return 0;
            else return -1;
        }
        if(n==2)
        {
            if(nums[0]==target) return 0;
            else if(nums[1]==target) return 1;
            else return -1;
        }
        int pivot=FindPivot(nums);
        int res1=-1,res2=-1;
        res1=BinarySearch(nums,0,pivot,target);
        if(pivot+1<n)
        {
            res2=BinarySearch(nums,pivot+1,n-1,target);
        }
        
        if(res1!=-1) return res1;
        else if(res2!=-1) return res2;
        else return -1;
       
    }
    int FindPivot(vector<int> &nums)
    {
        int n=nums.size();
        int left=0,right=n-1;
        while(left<right-1)
        {
            int a=(left+right)/2;
            if(nums[a]>nums[right])
            {
                left=a;
            }
            else
            {
                right=a;
            }
        }
        return left;
    }
    int BinarySearch(vector<int>&nums,int left,int right,int target)
    {
        int n=right-left+1;
        if(n==0) return -1;
        if(n==1)
        {
            if(nums[left]==target) return left;
            else return -1;
        }
        if(n==2)
        {
            if(nums[left]==target) return left;
            else if(nums[right]==target) return right;
            else return -1;
            
        }
        while(left<right-1)//当left和right相邻的时候跳出
        {
            int a=(left+right)/2;
            if(nums[a]<target)
            {
                left=a;
            }
            else if(nums[a]>target)
            {
                right=a;
            }
            else return a;
        }
        //与最后相邻的left和right比较,避免遗漏
        if(nums[left]==target) return left;
        else if(nums[right]==target) return right;
        else return -1;
    }
};

算法结果
Runtime: 4 ms, faster than 100.00% of C++ online submissions for Search in Rotated Sorted Array.
Memory Usage: 8.8 MB, less than 98.94% of C++ online submissions for Search in Rotated Sorted Array.

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值