给你一个整数数组 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
中的最大元素是唯一的
遍历数组找到最大值和次大值,最后判断是否满足条件即可。
class Solution:
def dominantIndex(self, nums: List[int]) -> int:
mx = mid = 0
ans = -1
for i, v in enumerate(nums):
if v > mx:
mid, mx = mx, v
ans = i
elif v > mid:
mid = v
return ans if mx >= 2 * mid else -1
class Solution {
public int dominantIndex(int[] nums) {
int mx = 0, mid = 0;
int ans = -1;
for (int i = 0; i < nums.length; ++i) {
if (nums[i] > mx) {
mid = mx;
mx = nums[i];
ans = i;
} else if (nums[i] > mid) {
mid = nums[i];
}
}
return mx >= mid * 2 ? ans : -1;
}
}
class Solution {
public:
int dominantIndex(vector<int>& nums) {
int mx = 0, mid = 0;
int ans = 0;
for (int i = 0; i < nums.size(); ++i) {
if (nums[i] > mx) {
mid = mx;
mx = nums[i];
ans = i;
} else if (nums[i] > mid)
mid = nums[i];
}
return mx >= mid * 2 ? ans : -1;
}
};
func dominantIndex(nums []int) int {
mx, mid := 0, 0
ans := 0
for i, v := range nums {
if v > mx {
mid, mx = mx, v
ans = i
} else if v > mid {
mid = v
}
}
if mx >= mid*2 {
return ans
}
return -1
}
/**
* @param {number[]} nums
* @return {number}
*/
var dominantIndex = function (nums) {
let mx = 0,
mid = 0;
let ans = 0;
for (let i = 0; i < nums.length; ++i) {
if (nums[i] > mx) {
mid = mx;
mx = nums[i];
ans = i;
} else if (nums[i] > mid) {
mid = nums[i];
}
}
return mx >= mid * 2 ? ans : -1;
};