题目
难度:★☆☆☆☆
类型:数组
在一个给定的数组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
如有疑问或建议,欢迎评论区留言~