在一个给定的数组nums中,总是存在一个最大元素 。
查找数组中的最大元素是否至少是数组中每个其他数字的两倍。
如果是,则返回最大元素的索引,否则返回-1。
示例 1:
输入: nums = [3, 6, 1, 0]
输出: 1
解释: 6是最大的整数, 对于数组中的其他整数,
6大于数组中其他元素的两倍。6的索引是1, 所以我们返回1.
示例 2:
输入: nums = [1, 2, 3, 4]
输出: -1
解释: 4没有超过3的两倍大, 所以我们返回 -1.
提示
- nums 的长度范围在[1, 50].
- 每个 nums[i] 的整数范围在 [0, 100].
golang语言版
func dominantIndex(nums []int) int {
n := len(nums)
if n == 1 {
return 0
}
// i1 是 nums 中第 1 大的数的索引号
// i2 是 nums 中第 2 大的数的索引号
i1, i2 := 0, 1
if nums[i1] < nums[i2] {
i1, i2 = i2, i1
}
for i := 2; i < n; i++ {
if nums[i1] < nums[i] {
i2, i1 = i1, i
} else if nums[i2] < nums[i] {
i2 = i
}
}
if nums[i2] == 0 || nums[i1]/nums[i2] >= 2 {
return i1
}
return -1
}