题目:
在一个给定的数组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, 99]
.
思路:分别找到数组中的最大值和第二大的值,判断最大值是否比第二大值的两倍大,如果是则返回最大值索引,如果不是则返回-1。
/**
* @param {number[]} nums
* @return {number}
*/
var dominantIndex = function(nums) {
let i = 0;
let j = 0;
let max = nums[0];
let maxIndex = 0;
if(nums.length==1)return 0;
for(i;i<nums.length;i++){
if(nums[i]>max) {
max = nums[i];
maxIndex = i;
}
}
let smax = 0;
for(j;j<nums.length;j++){
if(nums[j]>smax&&nums[j]<max) {
smax = nums[j];
}
}
if(max>=(2*smax))return maxIndex;
else return -1;
};
改进:
直接用索引比较,代码更整洁;缺点是用了两次循环遍历。
主要要考虑的2种情况是,max索引为0,数组长度为1。代码对这两种情况进行了单独判断。
/**
* @param {number[]} nums
* @return {number}
*/
var dominantIndex = function(nums) {
let max=0;
let smax=-1;
for(let i=0;i<nums.length;++i){
if(nums[i]>nums[max]){
max = i;
}
}
for(let j=0;j<nums.length;++j){
if(smax == -1 && nums[j]<nums[max]){
smax = j;
}
else if(nums[j]>nums[smax]&&nums[j]<nums[max]){
smax = j;
}
}
if(nums[max]>=(2*nums[smax])) return max;
else if(nums.length == 1) return 0;
else return -1;
};