在一个给定的数组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].
思路:
使用max函数取出最大值,使用index取出其索引后,删除最大元素。再次使用max函数取出现在的最大值,原第2大值,比较即可。
代码:
class Solution:
def dominantIndex(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if len(nums)==0:
return -1
if len(nums)==1:
return 0
m1=max(nums)
i1=nums.index(m1)
del nums[i1]
m2=max(nums)
if m1>=2*m2:
return i1
return -1
分析:
时间复杂度O(n),空间复杂度O(1)