天天看點

leetcode 154. 尋找旋轉排序數組中的最小值 IIleetcode 154. 尋找旋轉排序數組中的最小值 II

leetcode 154. 尋找旋轉排序數組中的最小值 II

題幹

已知一個長度為 n 的數組,預先按照升序排列,經由 1 到 n 次 旋轉 後,得到輸入數組。例如,原數組 nums = [0,1,4,4,5,6,7] 在變化後可能得到:

若旋轉 4 次,則可以得到 [4,5,6,7,0,1,4]

若旋轉 7 次,則可以得到 [0,1,4,4,5,6,7]

注意,數組 [a[0], a[1], a[2], …, a[n-1]] 旋轉一次 的結果為數組 [a[n-1], a[0], a[1], a[2], …, a[n-2]] 。

給你一個可能存在 重複 元素值的數組 nums ,它原來是一個升序排列的數組,并按上述情形進行了多次旋轉。請你找出并傳回數組中的 最小元素 。

示例 1:

輸入:nums = [1,3,5]

輸出:1

示例 2:

輸入:nums = [2,2,2,0,1]

輸出:0

提示:

n == nums.length

1 <= n <= 5000

-5000 <= nums[i] <= 5000

nums 原來是一個升序排序的數組,并進行了 1 至 n 次旋轉

進階:

這道題是 尋找旋轉排序數組中的最小值 的延伸題目。

允許重複會影響算法的時間複雜度嗎?會如何影響,為什麼?

知識點&算法

繼續二分,跟leetcode 153.尋找旋轉排序數組中的最小值的寫法基本一緻,注意三點相等時的區間縮小,導緻最差情況O(n)的時間複雜度

題解

class Solution {
public:
    int findMin(vector<int>& nums) {
        int n = nums.size();
        int l = 0 , r = n - 1 , ans = INT_MAX;
        while(l <= r){
            int m = (l + r) >> 1;
            ans = min(ans,nums[m]);
            if(nums[l] == nums[m] && nums[m] == nums[r]){
                l++;
                r--;
            }else if(nums[m] <= nums[r]){
                r = m - 1;
            }else if(nums[m] >= nums[l]){
                l = m + 1;
            }
        }
        return ans;
    }
};