链接
https://leetcode-cn.com/problems/largest-number-at-least-twice-of-others/description/
要求
在一个给定的数组nums中,总是存在一个最大元素 。
查找数组中的最大元素是否至少是数组中每个其他数字的两倍。
如果是,则返回最大元素的索引,否则返回-1。
输入: nums = [3, 6, 1, 0]
输出: 1
解释: 6是最大的整数, 对于数组中的其他整数,
6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.
输入: nums = [1, 2, 3, 4]
输出: -1
解释: 4没有超过3的两倍大, 所以我们返回 -1.
思路
- 先用if语句排除特殊情况。
- 用max取出最大值和位置,除以2之后和第2个最大值比较。
代码
执行用时:48 ms
class Solution:
def dominantIndex(self, nums):
if len(nums) == 1:
return 0
nums_max = [max(nums), nums.index(max(nums))]
nums.pop(nums_max[1])
nums_max_second = nums.pop(nums.index(max(nums)))
if nums_max[0] / 2 >= nums_max_second:
return nums_max[1]
else:
return -1