[파이썬 알고리즘]1464.leetcode - Maximum Product of Two Elements in an Array

문제 설명)

List에 들어있는 value 중 가장 큰 값과 2번째로 큰값을 각각 -1하고 곱하는 문제

 

아주 간단한 문제이다.

 

1) sort 이용

def Maximum(nums: list[int]) -> int:
    nums.sort()
    return (nums[-1] - 1) * (nums[-2] - 1)


a = [3, 4, 5, 2]
print(Maximum(a))

 

2) heapq 모듈의 nlargest 이용

import heapq


def Maximum(nums: list[int]) -> int:
    a, b = heapq.nlargest(2, nums)

    return (a - 1) * (b - 1)