题目

在一个给定的数组 nums 中，总是存在一个最大元素 。
查找数组中的最大元素是否至少是数组中每个其他数字的两倍。
如果是，则返回最大元素的索引，否则返回-1。

示例 1:**

  1. 输入: nums = [3, 6, 1, 0]
  2. 输出: 1
  3. 解释: 6是最大的整数, 对于数组中的其他整数,
  4. 6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.

示例 2:

输入: nums = [1, 2, 3, 4]
输出: -1
解释: 4没有超过3的两倍大, 所以我们返回 -1.

提示:

  • nums 的长度范围在 [1, 50].
  • 每个 nums[i] 的整数范围在 [0, 100].

解答

  1. 最大元素及其对应的索引
  2. 找到第二大的元素
  3. 判断是否满足条件
def dominantIndex(nums: [int]) -> int:
    first_max, first_max_index, second_max = nums[0], 0, 0
    for i in range(len(nums)):
        if nums[i] > first_max:
            second_max = first_max
            first_max = nums[i]
            first_max_index = i
            continue

        if nums[i] > second_max and nums[i] < first_max:
            second_max = nums[i]

    if first_max >= 2 * second_max:
        return first_max_index

    return -1
  • 循环条件里必须要有第二个 if 判断, 因为第二大的可能在第一大的后面;
  • 时间复杂度至少是其他数字两倍的最大数 - 图1

    地址

https://leetcode-cn.com/explore/learn/card/array-and-string/198/introduction-to-array/771/