[Leetcode] 747.至少是其他數字兩倍的最大數

題目:

在一個給定的數組nums中,總是存在一個最大元素 。

查找數組中的最大元素是否至少是數組中每個其他數字的兩倍。

如果是,則返回最大元素的索引,否則返回-1。

示例 1:

輸入: nums = [3, 6, 1, 0]
輸出: 1
解釋: 6是最大的整數, 對於數組中的其他整數,
6大於數組中其他元素的兩倍。6的索引是1, 所以我們返回1.

示例 2:

輸入: nums = [1, 2, 3, 4]
輸出: -1
解釋: 4沒有超過3的兩倍大, 所以我們返回 -1.

提示:

  1. nums 的長度範圍在[1, 50].
  2. 每個 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;
};

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章