Not Only Algorithm,不仅仅是算法,关注数学、算法、数据结构、程序员笔试面试以及一切涉及计算机编程之美的内容 。。
你的位置:NoAlGo博客 » 题解 » 

Leetcode Search in Rotated Sorted Array II

Leetcode algorithms 第 81 题:Search in Rotated Sorted Array II。

题目

Follow up for “Search in Rotated Sorted Array”:
What if duplicates are allowed?

Would this affect the run-time complexity? How and why?

Write a function to determine if a given target is in the array.

解答

和这题的第一版本类似,只是可能出现重复数字,而有了重复数字会使问题变得非常复杂。

例如对于数组 1 2 2 2 2 2 3,对第1个2进行翻转后,会得到2 2 2 2 3 1 2,这个数组首尾下标为0和6,得到中间位置的下标为3,于是首尾和中间位置的元素都相同,我们无法判断索要找的数字是在前半段还是在后半段,这种情况下,只能够从头开始以O(n)时间进行一次搜索。

如果不是以上的情况,还是可以按照第一版本的思路进行二分查找。

具体代码如下:

class Solution {
public:
    bool search(int A[], int n, int target) {
        if (n <= 0) return false;

        int st = 0, ed = n - 1;
        while (st <= ed)
        {
            int mid = st + ((ed - st) >> 1);
            if (target == A[mid]) return true;//找到
            
            if (A[mid] == A[st] && A[mid] == A[ed]) //首尾和中间都一样
            {
                for (int i = st; i <= ed; i++)
                    if (A[i] == target)
                        return true;
                return false;
            }
           
            if (A[mid] >= A[st]) //中间点在左半段
            {
                if (target >= A[st] && target < A[mid]) ed = mid - 1; //target在mid左边
                else st = mid + 1;
            }
            else                //中间点在右半段
            {
                if (target <= A[ed] && target > A[mid]) st = mid + 1; //target在mid右边
                else ed = mid - 1;
            }
        }
        return false;
    }
};
上一篇: 下一篇:

我的博客

NoAlGo头像编程这件小事牵扯到太多的知识,很容易知其然而不知其所以然,但真正了不起的程序员对自己程序的每一个字节都了如指掌,要立足基础理论,努力提升自我的专业修养。

站内搜索

最新评论