鏈接
https://leetcode-cn.com/problems/largest-number-at-least-twice-of-others/description/
要求
在一個(gè)給定的數(shù)組nums中，總是存在一個(gè)最大元素 。
查找數(shù)組中的最大元素是否至少是數(shù)組中每個(gè)其他數(shù)字的兩倍。
如果是，則返回最大元素的索引，否則返回-1。
輸入: nums = [3, 6, 1, 0]
輸出: 1
解釋: 6是最大的整數(shù), 對(duì)于數(shù)組中的其他整數(shù),
6大于數(shù)組中其他元素的兩倍。6的索引是1, 所以我們返回1.
輸入: nums = [1, 2, 3, 4]
輸出: -1
解釋: 4沒(méi)有超過(guò)3的兩倍大, 所以我們返回 -1.
思路
- 先用if語(yǔ)句排除特殊情況。
- 用max取出最大值和位置，除以2之后和第2個(gè)最大值比較。
代碼
執(zhí)行用時(shí)：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
