Question

747. 至少是其他数字两倍的最大数

English Version

题目描述

给你一个整数数组 nums ，其中总是存在 唯一的 一个最大整数 。

请你找出数组中的最大元素并检查它是否 至少是数组中每个其他数字的两倍 。如果是，则返回 最大元素的下标 ，否则返回 -1 。

 

示例 1：

输入：nums = [3,6,1,0]
输出：1
解释：6 是最大的整数，对于数组中的其他整数，6 至少是数组中其他元素的两倍。6 的下标是 1 ，所以返回 1 。

示例 2：

输入：nums = [1,2,3,4]
输出：-1
解释：4 没有超过 3 的两倍大，所以返回 -1 。

 

提示：

• 2 <= nums.length <= 50
• 0 <= nums[i] <= 100
• nums 中的最大元素是唯一的

解法

方法一：遍历

我们可以遍历数组 $nums$，找到数组中的最大值 $x$ 和第二大的值 $y$，如果 $x \ge 2y$，则返回 $x$ 的下标，否则返回 $-1$。

我们也可以先找到数组中的最大值 $x$，同时找到最大值 $x$ 的下标 $k$。然后再遍历一次数组，如果发现 $k$ 以外的元素 $y$ 满足 $x < 2y$，则返回 $-1$。否则遍历结束后返回 $k$。

时间复杂度 $O(n)$，其中 $n$ 是数组 $nums$ 的长度。空间复杂度 $O(1)$。

class Solution:
  def dominantIndex(self, nums: List[int]) -> int:
    x, y = nlargest(2, nums)
    return nums.index(x) if x >= 2 * y else -1

class Solution {
  public int dominantIndex(int[] nums) {
    int n = nums.length;
    int k = 0;
    for (int i = 0; i < n; ++i) {
      if (nums[k] < nums[i]) {
        k = i;
      }
    }
    for (int i = 0; i < n; ++i) {
      if (k != i && nums[k] < nums[i] * 2) {
        return -1;
      }
    }
    return k;
  }
}

class Solution {
public:
  int dominantIndex(vector<int>& nums) {
    int n = nums.size();
    int k = 0;
    for (int i = 0; i < n; ++i) {
      if (nums[k] < nums[i]) {
        k = i;
      }
    }
    for (int i = 0; i < n; ++i) {
      if (k != i && nums[k] < nums[i] * 2) {
        return -1;
      }
    }
    return k;
  }
};

func dominantIndex(nums []int) int {
  k := 0
  for i, x := range nums {
    if nums[k] < x {
      k = i
    }
  }
  for i, x := range nums {
    if k != i && nums[k] < x*2 {
      return -1
    }
  }
  return k
}

function dominantIndex(nums: number[]): number {
  let k = 0;
  for (let i = 0; i < nums.length; ++i) {
    if (nums[i] > nums[k]) {
      k = i;
    }
  }
  for (let i = 0; i < nums.length; ++i) {
    if (i !== k && nums[k] < nums[i] * 2) {
      return -1;
    }
  }
  return k;
}

/**
 * @param {number[]} nums
 * @return {number}
 */
var dominantIndex = function (nums) {
  let k = 0;
  for (let i = 0; i < nums.length; ++i) {
    if (nums[i] > nums[k]) {
      k = i;
    }
  }
  for (let i = 0; i < nums.length; ++i) {
    if (i !== k && nums[k] < nums[i] * 2) {
      return -1;
    }
  }
  return k;
};