Leetcode 561. 数组拆分 I
1 题目描述(Leetcode题目链接)
给定长度为 2n 的数组, 你的任务是将这些数分成 n 对, 例如 ( a 1 , b 1 ) , ( a 2 , b 2 ) , . . . , ( a n , b n ) (a1, b1), (a2, b2), ..., (an, bn) (a1,b1),(a2,b2),...,(an,bn) ,使得从1 到 n 的 m i n ( a i , b i ) min(ai, bi) 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 ) O(n) 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;
}