題目

難度：★☆☆☆☆
類型：數組

在一個給定的數組nums中，總是存在一個最大元素 。

查找數組中的最大元素是否至少是數組中每個其他數字的兩倍。

如果是，則返回最大元素的索引，否則返回-1。

提示
nums 的長度范圍在[1, 50].
每個 nums[i] 的整數范圍在 [0, 99].

示例

示例 1
輸入: nums = [3, 6, 1, 0]
輸出: 1
解釋: 6是最大的整數, 對于數組中的其他整數,
6大于數組中其他元素的兩倍。6的索引是1, 所以我們返回1.

示例 2
輸入: nums = [1, 2, 3, 4]
輸出: -1
解釋: 4沒有超過3的兩倍大, 所以我們返回 -1.

解答

我們首先準備三個變量并初始化：
最大數（max_1）：-inf
次大數（max_2）：-inf
和最大數的下標（max_id）：-1

在遍歷過程中，我們不斷更新這三個變量，在最后判斷最大數與次大數的2倍之間的差，并根據判斷結果返回相應參數即可。

class Solution:
    def dominantIndex(self, nums):
        if not nums:
            return -1

        max_1 = max_2 = -float('inf')
        max_id = -1
        for i in range(len(nums)):
            if nums[i] > max_1:
                max_1, max_2, max_id = nums[i], max_1, i
            elif max_2 < nums[i] < max_1:
                max_2 = nums[i]

        return max_id if max_1 >= max_2 * 2 else -1

如有疑問或建議，歡迎評論區(qū)留言~