Largest Number At Least Twice of Others in Python

Suppose we have an integer array called nums, now there is always exactly one largest element. We have to check whether the largest element in the array is at least twice as much as every other number in the array. If it is so, then we have to find the index of the largest element, otherwise return -1.

So, if the input is like [3,6,1,0], then the output will be 1, as 6 is the largest number, and for every other number in the array x, 6 is more than twice as big as x. As the index of 6 is 1, then the output is also 1.

To solve this, we will follow these steps −

• maximum := maximum of nums
• for i in range 0 to size of nums, do
  • if nums[i] is same as maximum, then
    • maxindex := i
  • if nums[i] is not same as maximum and maximum < 2*(nums[i]), then
    • return -1

Let us see the following implementation to get better understanding −

Example

 Live Demo

class Solution:
   def dominantIndex(self, nums):
      maximum = max(nums)
      for i in range(len(nums)):
         if nums[i] == maximum:
            maxindex = i
         if nums[i] != maximum and maximum < 2*(nums[i]):
            return -1
      return maxindex
ob = Solution()
print(ob.dominantIndex([3, 6, 1, 0]))

Input

[3, 6, 1, 0]

Output

1

Arnab Chakraborty

Published on 04-Jul-2020 13:27:57