Leetcode 561. 数组拆分 I（Array Partition I）

Leetcode 561. 数组拆分 I

1 题目描述（Leetcode题目链接）

  给定长度为 2n 的数组, 你的任务是将这些数分成 n 对, 例如 $(a1,b1),(a2,b2),...,(an,bn)$ ，使得从1 到 n 的 $min(ai,bi)$ 总和最大。

输入: [1,4,3,2]

输出: 4
解释: n 等于 2, 最大总和为 4 = min(1, 2) + min(3, 4).

提示:

• n 是正整数,范围在 [1, 10000].
• 数组中的元素范围在 [-10000, 10000].

2 题解

  排序后按照顺序两组一对。这个题目考察的不是排序。

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

另外由于数组中的元素规定了范围，所以可以使用桶排序达到$O(n)$的时间复杂度。下面是官方题解。

public class Solution {
    public int arrayPairSum(int[] nums) {
        int[] arr = new int[20001];
        int lim = 10000;
        for (int num: nums)
            arr[num + lim]++;
        int d = 0, sum = 0;
        for (int i = -10000; i <= 10000; i++) {
            sum += (arr[i + lim] + 1 - d) / 2 * i;
            d = (2 + arr[i + lim] - d) % 2;
        }
        return sum;
    }