題目:
在一個給定的數組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;
};