leetcode561题数组拆分(时间复杂度为O(n)的解法)
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/array-partition-i
题目描述
给定长度为 2n 的数组, 你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), …, (an, bn) ,使得从1 到 n 的 min(ai, bi) 总和最大。
示例 1:
输入: [1,4,3,2]
输出: 4
解释: n 等于 2, 最大总和为 4 = min(1, 2) + min(3, 4).
提示:
n 是正整数,范围在 [1, 10000].
数组中的元素范围在 [-10000, 10000].
基于排序的解法,找出第奇数个小的元素
时间复杂度O(nlogn)
public static int arrayPairSum(int[] nums) {
Arrays.sort(nums);
int sum = 0;
for (int i = 0; i < nums.length; i++) {
if (i % 2 == 0)
sum += nums[i];
}
return sum;
}
不需要排序,题目给出数组中的元素范围在 [-10000, 10000],可用一个数组保存元素的位置arr[nums[i] + 10000]++,再找出第奇数个元素求和
时间复杂度O(n)
public static int arrayPairSum2(int[] nums) {
int sum = 0;
int[] arr = new int[20001];
for (int i = 0; i < nums.length; i++) {
arr[nums[i] + 10000]++;
}
int n = 1;
for (int j = 0; j < arr.length; j++) {
while (arr[j]-- != 0) {
if (n % 2 != 0) {
sum += j - 10000;
n++;
}
else
n++;
}
}
return sum;
}