[Python]LeetCode 747. 至少是其他数字两倍的最大数(解法一)

298 阅读1分钟

在一个给定的数组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].

  • 思路:
1. 分析题意只需找出最大数比次大数大2倍即可
2. 注意2个例外情况
    2.1 1个元素的情况,返回自己
    2.2 如果有多个最大数,那么是排重的

3. 综上,可以用集合来做,取最大数,去掉最大数,再取最大数(次大数),比较即可。
  • 示例代码

class Solution:
    def dominantIndex(self, nums: List[int]) -> int:
        if len(nums) == 1:
            return 0
        else:
            set_nums = set(nums)
            max_num = max(set_nums)
            set_nums.remove(max_num)
            max_2_num = max(set_nums)
            if max_num >= max_2_num*2:
                return nums.index(max_num)
            else:
                return -1

  • 执行情况,不算优解,也不差,很普通的解法。