Leetcode_747 Largest Number At Least Twice of Others

发表于 | 分类于

在一个给定的数组nums中，总是存在一个最大元素 。

查找数组中的最大元素是否至少是数组中每个其他数字的两倍。

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

示例 1:

输入: nums = [3, 6, 1, 0]
输出: 1
解释: 6是最大的整数, 对于数组中的其他整数,
6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.

示例 2:

输入: nums = [1, 2, 3, 4]
输出: -1
解释: 4没有超过3的两倍大, 所以我们返回 -1.

提示:

nums 的长度范围在[1, 50].
每个 nums[i] 的整数范围在 [0, 99].

“””

分析：
查找数组中的最大元素是否至少是数组中每个其他数字的两倍，

设置临时变量保存数组最大值与第二大值，当最大值大于第二大值的两倍即满足，

否则不满足条件

“””

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
class Solution:
    def dominantIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        first, second, index = 0, 0, 0
        for i in range(len(nums)):
            if nums[i] > first:
                second = first
                first = nums[i]
                index = i
            elif nums[i] > second:
                second = nums[i]

        return index if (second == 0 or first / second >= 2) else -1