给你一个整数数组 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 至少是其他数字的两倍。
提示:
1 <= nums.length <= 50
0 <= nums[i] <= 100
nums 中的最大元素是唯一的
思路与算法
遍历数组分别找到数组的最大值 mac和次大值 less, 如果 max ≥ less×2 成立,则最大值至少是数组其余数字的两倍,此时返回最大值的下标,否则返回 −1。
为了返回最大值的下标,我们需要在计算最大值的同时记录最大值的下标。
class Solution {
public int dominantIndex(int[] nums) {
// 只有一个数字,返回当前数字的下标
if (nums.length == 1) {
return 0;
}
int max = 0, idx = 0, less = 1;
for (int i = 0; i < nums.length; i++) {
if (nums[i] > max) {
less = max;
max = nums[i];
idx = i;
} else if (nums[i] > less) {
less = nums[i];
}
}
return max >= (less * 2) ? idx : -1;
}
}
复杂度分析
时间复杂度:O(N),其中 N 是数组的长度。遍历整个数组需要 O(N)。
空间复杂度:O(1)。