题目描述
在一个给定的数组nums中,总是存在一个最大元素 。
查找数组中的最大元素是否至少是数组中每个其他数字的两倍。
如果是,则返回最大元素的索引,否则返回-1。
示例 1:
输入: nums = [3, 6, 1, 0]
输出: 1
解释: 6是最大的整数, 对于数组中的其他整数,
6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.
示例 2:
输入: nums = [1, 2, 3, 4]
输出: -1
解释: 4没有超过3的两倍大, 所以我们返回 -1.
思路分析
- 第一个for循环,找出nums列表中最大的数,并记录其索引
- 第二个for循环,如果找到的数不是最大的数,且这个数的两倍大于最大的数,那么就按题目要求返回-1
- 如果遍历全部,没有找到,返回第一次遍历找到的max_index
代码示例
class Solution(object):
def dominantIndex(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
max_num = 0
max_index = 0
for i in range(len(nums)):
if nums[i] > max_num:
max_num = nums[i]
max_index = i
for i,j in enumerate(nums):
if j != max_num and 2*j > max_num:
return -1
return max_index