154. Find Minimum in Rotated Sorted Array II

来源:互联网 发布:lamp一键安装包ubuntu 编辑:程序博客网 时间:2024/06/07 13:25

Follow up for "Find Minimum in Rotated Sorted Array":
What if duplicates are allowed?

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

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

Find the minimum element.

The array may contain duplicates.

这道题与上周那道的唯一区别在于元素可能有重复,我们仍然采用上面的方法,只是需要处理mid与start相等这种
额外情况。
nums[mid] > nums[start],右半区间查找。
nums[mid] < nums[start],左半区间查找。
nums[mid] = nums[start],出现这种情况,我们跳过start,重新查找,譬如[2,2,2,1],nums[mid] = nums[start]都为2,
这时候我们跳过start,使用[2,2,1]继续查找。

代码如下:

class Solution {
public:
    int findMin(vector<int>& nums) {
        int size = nums.size();
        if(size == 0) {
            return 0;
        } else if(size == 1) {
            return nums[0];
        } else if(size == 2) {
            return min(nums[0], nums[1]);
        }
        
        
        int start = 0;
        int stop = size - 1;
        while(start < stop - 1) {
            if(nums[start] < nums[stop]) {
                return nums[start];
            }
            int mid = start + (stop - start) / 2;
            if(nums[mid] > nums[start]) {
                start = mid;
            }else if(nums[mid]<nums[start]){
                stop =mid;
            }else{
                start++;
            }
        }
        return min(nums[start], nums[stop]);
            
        
    }
};

0 0