【LeetCode】至少是其他数字两倍的最大数

题目描述

747.给你一个整数数组 nums ，其中总是存在 唯一的 一个最大整数 。

请你找出数组中的最大元素并检查它是否 至少是数组中每个其他数字的两倍 。如果是，则返回 最大元素的下标 ，否则返回 -1

示例

示例 1：

输入：nums = [3,6,1,0]
输出：1
解释：6 是最大的整数，对于数组中的其他整数，6 大于数组中其他元素的两倍。6 的下标是 1 ，所以返回 1 。
示例 2：

输入：nums = [1,2,3,4]
输出：-1
解释：4 没有超过 3 的两倍大，所以返回 -1 。
示例 3：

输入：nums = [1]
输出：0
解释：因为不存在其他数字，所以认为现有数字 1 至少是其他数字的两倍。

方法

找到最大值及其下标和次大值，如果次大值的两倍小于最大值则返回最大值下标，否则返回-1.对于一个数的情况，返回0。

总代码

class Solution:
    def dominantIndex(self, nums: List[int]) -> int:
        n = len(nums)
        if n < 2:
            return 0
        max_val = max(nums)
        for i in range(n):
            if nums[i] == max_val:
                nums[i] = 0
                break
        second_max = max(nums)        
        if second_max * 2 <= max_val:
            return i
        else:
            return -1