要求
给你一个整数数组 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 至少是其他数字的两倍。
核心代码
class Solution:
def dominantIndex(self, nums: List[int]) -> int:
i = nums.index(max(nums))
nums.sort()
if len(nums) == 1:
return 0
if nums[-1] >= nums[-2] * 2:
return i
return -1
解题思路:我们先找到最大值的坐标保存下来,然后我们对数据列表进行排序,当我们的最大值大于等于次大值的两倍,也就是大于等于数据列表中其他所有值的两倍,所以返回下标即可,要不然,就是不存在返回-1即可。