개요

LeetCode #561, Array Partition I 문제를 풀어봅니다.

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

Sorting을 해 주면 2개씩 pair를 잡을 때 각 pair에서 작은 값들은 짝수 인덱스에 위치합니다. 2개 pair의 최솟값 합을 가장 크게 하려면 sorting 후 앞에서부터 2개씩 끊는 경우입니다. arrayPairSum 함수는 nums 리스트의 인덱스로 직접 접근했고, arrayPairSum_ver2 함수는 sorting 후 짝수 인덱스만 잘라서 뽑습니다.

class Solution:
    def arrayPairSum(self, nums: List[int]) -> int:
        # After sorting, when two pairs are held, the small value is in the even index
        nums.sort()
        sums = 0
        for i in range(0, len(nums), 2):
            sums += nums[i]
            
        return sums
    
    def arrayPairSum_ver2(self, nums: List[int]) -> int:
        return sum(sorted(nums)[::2])