1. 程式人生 > >LeetCode 747. 至少是其他數字兩倍的最大數(C、C++、python)

LeetCode 747. 至少是其他數字兩倍的最大數(C、C++、python)

在一個給定的陣列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].

C

int dominantIndex(int* nums, int numsSize) 
{
    int n=numsSize;
    int max=nums[0];
    int idx=0;
    for(int i=0;i<n;i++)
    {
        if(nums[i]>max)
        {
            max=nums[i];
            idx=i;
        }
    }
    for(int i=0;i<n;i++)
    {
        if(i!=idx && nums[idx]<2*nums[i])
        {
            return -1;
        }
    }
    return idx;
}

C++

class Solution {
public:
    int dominantIndex(vector<int>& nums) 
    {
        int index=max_element(nums.begin(),nums.end())-nums.begin();
        int n=nums.size();
        for(int i=0;i<n;i++)
        {
            if(i!=index && nums[index]<2*nums[i])
            {
                return -1;
            }
        }
        return index;
    }    
};

python

class Solution:
    def dominantIndex(self, nums):
        """
        :type nums: List[int]
        :rtype: int
        """
        idx=nums.index(max(nums))
        for i in range(len(nums)):
            if i!=idx and nums[idx]<2*nums[i]:
                return -1
        return idx