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 。
示例3:
输入:nums = [1]
输出:0
解释:因为不存在其他数字,所以认为现有数字 1 至少是其他数字的两倍。
解题思路:
遍历数组分别找到数组的最大值 b 和次大值 a。如果 b ≥ a * 2 成立,则最大值至少是数组其余数字的两倍,此时返回最大值的下标,否则返回 -1。 为了返回最大值的下标,我们需要在计算最大值的同时记录最大值的下标。
Java代码:
class Solution {
public int dominantIndex(int[] nums) {
if (nums.length == 1) return 0;
int a = -1, b = -1;
int idx = -1;
for (int i = 0; i < nums.length; i++) {
if (nums[i] > b) {
a = b;
b = nums[i];
idx = i;
} else if (nums[i] > a) {
a = nums[i];
}
}
return b >= a * 2 ? idx : -1;
}
}
Go代码:
func dominantIndex(nums []int) int {
if len(nums) == 1 {
return 0;
}
a, b, idx := -1, -1, -1
for i, num := range nums {
if num > b {
a, b, idx = b, num, i
} else if num > a {
a = num
}
}
if b >= a * 2 {
return idx
}
return -1;
}
复杂度分析:
- 时间复杂度:O(n),其中 n 为 nums 数组的长度
- 空间复杂度:O(1),只使用了有限几个变量