LeetCode 747. 至少是其他數字兩倍的最大數(C、C++、python)
阿新 • • 發佈:2018-11-21
在一個給定的陣列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