[LeetCode] 561. Array Partition 1

Jadon·2022년 1월 3일
0
post-thumbnail

Given an integer array nums of 2n integers, group these integers into n pairs (a1, b1), (a2, b2), ..., (an, bn) such that the sum of min(ai, bi) for all i is maximized. Return the maximized sum.

Example 1:

Input: nums = [1,4,3,2]
Output: 4
Explanation: All possible pairings (ignoring the ordering of elements) are:
1. (1, 4), (2, 3) -> min(1, 4) + min(2, 3) = 1 + 2 = 3
2. (1, 3), (2, 4) -> min(1, 3) + min(2, 4) = 1 + 2 = 3
3. (1, 2), (3, 4) -> min(1, 2) + min(3, 4) = 1 + 3 = 4
So the maximum possible sum is 4.

Example 2:

Input: nums = [6,2,6,5,1,2]
Output: 9
Explanation: The optimal pairing is (2, 1), (2, 5), (6, 6). min(2, 1) + min(2, 5) + min(6, 6) = 1 + 2 + 6 = 9.

Constraints:

1 <= n <= 104
nums.length == 2 * n
-104 <= nums[i] <= 104

My Solution

숫자들을 페어로 묶은 다음 작은 값의 합을 최대로 만드는 문제이다.
작은 값은 작은 값끼리, 큰 값은 큰 값끼리 묶어야 min(a, b) + min(c, d)와 같은 연산을 할 때 합이 가장 커지는 것을 알 수 있다.

  1. 정렬을 한다.
  2. 결국 2개씩 페어로 묶으면 왼쪽 값이 더해지는 과정이기에 짝수 index의 값만 가져와서 더하면 된다.
class Solution:
    def arrayPairSum(self, nums: List[int]) -> int:
          return sum(sorted(nums)[::2])

0개의 댓글