题目描述
给你一个整数数组 nums
,其中总是存在 唯一的 一个最大整数 。
请你找出数组中的最大元素并检查它是否 至少是数组中每个其他数字的两倍。如果是,则返回 最大元素的下标,否则返回 -1
。
样例
输入:nums = [3,6,1,0]
输出:1
解释:6 是最大的整数,对于数组中的其他整数,6 大于数组中其他元素的两倍。
6 的下标是 1,所以返回 1。
输入:nums = [1,2,3,4]
输出:-1
解释:4 没有超过 3 的两倍大,所以返回 -1。
输入:nums = [1]
输出:0
解释:因为不存在其他数字,所以认为现有数字 1 至少是其他数字的两倍。
限制
1 <= nums.length <= 50
0 <= nums[i] <= 100
nums
中的最大元素是唯一的。
算法
(模拟) $O(n)$
- 遍历一次数组找到最大值,再遍历一次数组判定。
时间复杂度
- 遍历两次数组,故总时间复杂度为 $O(n)$。
空间复杂度
- 仅需要常数的额外空间。
C++ 代码
class Solution {
public:
int dominantIndex(vector<int>& nums) {
const int n = nums.size();
int idx = 0;
for (int i = 1; i < n; i++)
if (nums[i] > nums[idx])
idx = i;
for (int i = 0; i < n; i++)
if (i != idx && nums[idx] < 2 * nums[i])
return -1;
return idx;
}
};