【题目描述】
在一个给定的数组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, 100].
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/largest-number-at-least-twice-of-others
【解题思路】
要找到最大元素是否至少是数组中每个其他数字的两倍,只需要判断最大元素是否至少是第二大元素的两倍即可。用一次扫描找出最大和第二大元素,需要注意的是两个边界条件:数组长度为1 和 第二大元素为0.
用Python实现的代码如下:
class Solution(object):
def dominantIndex(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
if len(nums)==1:
return 0
first, second = -1, -1
first_pos, second_pos = -1, -1
for i in range(len(nums)):
if nums[i] > first:
second = first
second_pos = first_pos
first = nums[i]
first_pos = i
elif second < nums[i] <= first:
second = nums[i]
second_pos = i
if second==0 or first/second>=2:
return first_pos
else:
return -1