LeetCode(No.747)--至少是其他數字兩倍的最大數

在一個給定的數組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].

思路: 直接找出最大值的下標,然後對數組從小到大進行排序,如果最後一個數是倒數第二個數的兩倍以上,那麼返回起先找到的最大值的下標,否則返回-1

class Solution:
    def dominantIndex(self, nums: List[int]) -> int:
        i = nums.index(max(nums))
        nums.sort()
        if len(nums) == 1:
            return 0
        if nums[-1] >= 2 * nums[-2]:
            return i
        return -1

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章