LintCode 尋找旋轉排序陣列中的最小值 II

weixin_33976072發表於2017-03-16

題目

假設一個旋轉排序的陣列其起始位置是未知的(比如0 1 2 4 5 6 7 可能變成是4 5 6 7 0 1 2)。

你需要找到其中最小的元素。

陣列中可能存在重複的元素。

樣例
給出[4,4,5,6,7,0,1,2] 返回 0

分析

這次可以出現重複元素,很簡單,只要判斷的時候加等於的判斷即可,如果mid等於end,那麼end--最小值肯定還在其中。

程式碼

public class Solution {
    /**
     * @param num: a rotated sorted array
     * @return: the minimum number in the array
     */
    public int findMin(int[] nums) {
        // write your code here
        if (nums == null || nums.length == 0) {
            return -1;
        }
        
        int start = 0, end = nums.length - 1;
        while (start + 1 < end) {
            int mid = start + (end - start) / 2;
            if(nums[mid]>nums[start] && nums[mid] < nums[end])
                return nums[0];
            else if (nums[mid] == nums[end]) {
                // if mid equals to end, that means it's fine to remove end
                // the smallest element won't be removed
                end--;
            } else if (nums[mid] < nums[end]) {
                end = mid;
                // of course you can merge == & <
            } else {
                start = mid;
                // or start = mid + 1
            }
        }
        
        if (nums[start] <= nums[end]) {
            return nums[start];
        }
        return nums[end];
    }
}

相關文章