[리트코드/파이썬] 561. Array Partition(그리디)
난이도: Easy
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
[내 풀이]
※ 소요시간 : 2 분
※ 풀이 전략
- 두 수를 한 쌍으로 묶었을 때 한 쌍 중 최소 값들의 합이 최대가 되게 쌍을 묶는 문제.
- 가장 작은 값부터 처리해야하므로 우선 정렬을 한다.
- 가장 작은 값A과 묶이는 값B는 A를 제외한 nums 중에서 가장 최솟값이어야 한다: nums를 정렬한 경우 바로 다음 값이 묶인다.
- 따라서 결론적으로 정렬한 nums에서 최솟값들은 0번째 부터 step을 2씩 준 값들이다. 이 값들을 합하면 구하는 값이다.
class Solution:
def arrayPairSum(self, nums: List[int]) -> int:
nums.sort()
return sum([x for x in nums[::2]])
※ 중요 포인트
- 그리디
댓글