题目:
在一个给定的数组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].
分析:
首先分析题目的逻辑,最大元素至少是其他每个数的两倍,-》至少是其他数字中 第二大的数字的两倍;
所以只要遍历一次数组,找到最大的和第二大的数字,再去判断是否为2倍即可;
遍历过程中,使用两个变量分别存储最大的数和第二大的数;
注意
需要注意,判断的当前数字 < 当前最大数字后,还有判别是否大于 第二大的数字,及时更新第二大数字;
代码:
class Solution {
public:
int dominantIndex(vector<int>& nums) {
int n = nums.size();
if(n == 1) return 0;
int ret = 0, max_num = 0, second_num = 0;
for(int i = 0;i < n; i++) // 对于每一个元素
{
if(nums[i] > max_num){ //如果当前元素 > 最大元素
second_num = max_num;
ret = i;
max_num = nums[i];
}
else if(nums[i] < max_num && nums[i] > second_num){ // 如果小于最大,大于第二大
second_num = nums[i];
}
}
if(second_num == 0)
return ret;
else if(max_num / second_num >= 2)
return ret;
else
return -1;
}
};