DEV Community

Abhishek Chaudhary
Abhishek Chaudhary

Posted on

Largest Number At Least Twice of Others

You are given an integer array nums where the largest integer is unique.

Determine whether the largest element in the array is at least twice as much as every other number in the array. If it is, return the index of the largest element, or return -1 otherwise.

Example 1:

Input: nums = [3,6,1,0]
Output: 1
Explanation: 6 is the largest integer.
For every other number in the array x, 6 is at least twice as big as x.
The index of value 6 is 1, so we return 1.

Example 2:

Input: nums = [1,2,3,4]
Output: -1
Explanation: 4 is less than twice the value of 3, so we return -1.

Example 3:

Input: nums = [1]
Output: 0
Explanation: 1 is trivially at least twice the value as any other number because there are no other numbers.

Constraints:

  • 1 <= nums.length <= 50
  • 0 <= nums[i] <= 100
  • The largest element in nums is unique.

SOLUTION:

class Solution:
    def dominantIndex(self, nums: List[int]) -> int:
        n = len(nums)
        if n < 2:
            return 0
        currmax = float('-inf')
        secmax = float('-inf')
        for num in nums:
            if num > currmax:
                secmax = currmax
                currmax = num
            elif num > secmax and num != currmax:
                secmax = num
        return nums.index(currmax) if currmax >= 2 * secmax else -1
Enter fullscreen mode Exit fullscreen mode

Top comments (0)