LeetCode 154. Find Minimum in Rotated Sorted Array II

来源:互联网 发布:网站js被劫持 编辑:程序博客网 时间:2024/05/16 19:08

题目

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.


思路

一个有序数组的开始一部分元素被搬到数组的末尾,是旋转有序数组,题目要求找到旋转数组的最小值。这道题的关键在于这个被旋转的有序数组中是有重复的元素。
最直接的解法就是从头到尾遍历一遍数组,找到最小的元素,这是一个O(n)的方法。
代码如下:

class Solution {public:    int findMin(vector<int>& nums) {        int min = nums[0];        for(int i = 1;i < nums.size();i++)        {            if(nums[i] < min)                min = nums[i];        }        return min;    }};

但这个思路并没有用到输入数组是一个旋转数组。对于这个旋转数组其实可以被分为两个有序的子数组,而两个子数组中间的数就是最小的数。所以可以类似于二分查找,分别有两个指针指向数组第一个元素和最后一个元素,按照题目的说法,第一个元素肯定大于最后一个元素。
找到数组最中间的数,跟最后一个元素做对比有三种情况:
1)如果最后一个元素小于最中间的数,则说明最中间的数在第一个递增子序列中,将第二个指针移到中间。
2)如果最后一个元素大于中间的数,则说明中间的数在第二个子序列中,则将第一个指针移动到中间。
3)如果最后一个元素等于中间的数,这时如果第一个元素也等于中间的数,则需要顺序查找,分别移动两个指针向中间靠拢。如果中间的数不等于第一个元素,则将第二个指针移到中间。

移动两个指针的时候,都会将查找范围缩小到原来的一半。


代码

class Solution {public:    int findMin(vector<int>& nums) {        if(nums.size() == 1)            return nums[0];        int index1 = 0;        int index2 = nums.size()-1;        int mid = index1;        while(index1<index2)        {            mid = (index1+index2)/2;            if(nums[mid] > nums[index2])                index1 = mid+1;            else if(nums[mid] < nums[index2])                index2 = mid;            else            {                if(nums[index1] == nums[mid])                {                    index1++;                    index2--;                }                else                     index2 = mid;            }        }        return nums[index2];    }};
0 0
原创粉丝点击