747. 至少是其他数字两倍的最大数（Python）

题目

难度：★☆☆☆☆
类型：数组

在一个给定的数组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

如有疑问或建议，欢迎评论区留言~