leetcode 154. 寻找旋转排序数组中的最小值 II (二分)

假设按照升序排序的数组在预先未知的某个点上进行了旋转。

( 例如,数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。

请找出其中最小的元素。

注意数组中可能存在重复的元素。

示例 1:

输入: [1,3,5]
输出: 1
示例 2:

输入: [2,2,2,0,1]
输出: 0

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/find-minimum-in-rotated-sorted-array-ii
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

好像是一到很经典的题目。做法很容易想到,显然还是二分,不过怎么二分就是难点了。
首先我们只需要考虑有哪些情况

  • 1 2 3 4 5
  • 3 4 5 1 2
  • 2 2 2 1 2
  • 2 1 2 2 2
  • 2 2 2 2 2
    大概就这五种情况,我第一次写的时候忘记考虑第四种情况了,结果错了。。。
    首先我们可以确定的是第一个数一定大于或等于最后一个数,因为最开始是一个递增的数组旋转的。
    对于二分,我们中间值的判断条件是nums[l] < nums[mid]时,肯定l = mid.因为前面一定时递增的(因为后面部分的最大值也是小于前面部分的最小值的)(注意思考)。然后时nums[l] == nums[mid]时,这个情况不能和第一种混为一谈,考虑第四种样例,我们只能是执行l++(不过这时候复杂度就退化了。。).然后就是else r = mid;(其实就是nums[mid] <= nums[r])这样就可以写出完整的代码了。
class Solution {
public:
    int findMin(vector<int>& nums) {
        if(nums.size() == 0)    return 0;
        
        int l = 0,r = nums.size()-1;
        while(nums[l] >= nums[r]){
            if(l+1 == r || l == r)    break;//注意这里一定要特判,因为只有一个或两个元素时,会陷入死循环
            int mid = (l+r)/2;
            if(nums[l] < nums[mid])
                l = mid;
            else if(nums[l] == nums[mid])
                l++;
            else
                r = mid;
        }
        return min(nums[l],nums[r]);//因为无法保证nums[l]一定小
    }
};
©️2020 CSDN 皮肤主题: 编程工作室 设计师:CSDN官方博客 返回首页