leetcode 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.

这道题是二分查找Search Insert Position的变体,看似有点麻烦,其实理清一下还是比较简单的。因为rotate的缘故,当我们切取一半的时候可能会出现误区,所以我们要做进一步的判断。具体来说,假设数组是A,每次左边缘为start,右边缘为end,还有中间位置是mid。在每次迭代中,分三种情况:
(1)如果target==A[mid],那么mid就是我们要的结果,直接返回;
(2)如果A[mid]>=A[start],那么说明从start到mid一定是有序的(没有受到rotate的影响),那么我们只需要判断target是不是在start到mid之间,如果是则把右边缘移到m-1,否则就target在另一半,即把左边缘移到m+1。
(3)如果A[m]<A[start],那么说明从mid到end一定是有序的,同样只需要判断target是否在这个范围内,相应的移动边缘即可。

根据以上方法,每次我们都可以切掉一半的数据,所以算法的时间复杂度是O(logn),空间复杂度是O(1)。代码如下:

class Solution {
public:
    int search(int A[], int n, int target) {
        int start = 0;
        int end = n-1;
        while(start<= end) {
            int mid = (start+end)/2;
            if(A[mid] == target)
            return mid;
            if(A[mid]>=A[start]) {
                if(A[start]<= target && A[mid]> target) {
                    end = mid-1;
                } else {
                    start = mid +1;
                }
            } else {
                if(A[mid]< target && A[end]>= target) {
                    start = mid + 1;
                } else {
                    end = mid -1;
                }
            }
        }
        return -1;
    }
};

Search in Rotated Sorted Array唯一的区别是这道题目中元素会有重复的情况出现。不过正是因为这个条件的出现,出现了比较复杂的case,甚至影响到了算法的时间复杂度。原来我们是依靠中间和边缘元素的大小关系,来判断哪一半是不受rotate影响,仍然有序的。而现在因为重复的出现,如果我们遇到中间和边缘相等的情况,我们就丢失了哪边有序的信息,因为哪边都有可能是有序的结果。假设原数组是{1,2,3,3,3,3,3},那么旋转之后有可能是{3,3,3,3,3,1,2},或者{3,1,2,3,3,3,3},这样的我们判断左边缘和中心的时候都是3,如果我们要寻找1或者2,我们并不知道应该跳向哪一半。解决的办法只能是对边缘移动一步,直到边缘和中间不在相等或者相遇,这就导致了会有不能切去一半的可能。所以最坏情况(比如全部都是一个元素,或者只有一个元素不同于其他元素,而他就在最后一个)就会出现每次移动一步,总共是n步,算法的时间复杂度变成O(n)。代码如下:

class Solution {
public:
    bool search(int A[], int n, int target) {
        int start = 0;
        int end = n-1;
        while(start<= end) {
            int mid = (start+end)/2;
            if(A[mid] == target)
            return true;
            if(A[mid]>A[start]) {
                if(A[start]<= target && A[mid]> target) {
                    end = mid-1;
                } else {
                    start = mid +1;
                }
            } else if(A[mid]<A[start]){
                if(A[mid]< target && A[end]>= target) {
                    start = mid + 1;
                } else {
                    end = mid -1;
                }
            } else {
                start++;
            }
        }
        return false;
    }
};



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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值