[코테] 리트코드 배열 561. Array Partition

561. Array Partition

문제 링크

https://leetcode.com/problems/array-partition/description/

문제 설명

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.

제한 사항

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

입출력 예 #1

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

입출력 예 #2

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

문제 풀이

리스트를 오름차순으로 정렬한 이후 연속되는 숫자의 min값을 구하는 문제이므로 0, 2, 4, … 짝수 번째 index를 가진 숫자의 합을 구하면 된다.

class Solution:
    def arrayPairSum(self, nums: List[int]) -> int:
        return sum(sorted(nums)[::2])