数据结构习题–数组拆分
给定长度为 2n 的整数数组 nums ,你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), …, (an, bn) ,使得从 1 到 n 的 min(ai, bi) 总和最大。
方法一:直接遍历
分析
我们分析其实,要使每个组合的最小值相加的和最大,那么问题在于每个组合的最小值,我们通过对数组进行排序,取每两个数的前一个数,进行累加和,最后得到总和
代码
import java.util.Arrays;
public class demergeArray {
public int arrayPairSum(int[] nums) {
Arrays.sort(nums);
int sum = 0;
for (int i = 0; i < nums.length; i += 2) {
sum += nums[i];
}
return sum;
}
}
复杂度
时间复杂度:O(nlogn),即为对数组 nums 进行排序的时间复杂度。
空间复杂度:O(logn),即为排序需要使用的栈空间
方法二:计数排序
分析
先统计每个数出现的次数,把每个数的值转换为一个新数组的索引,然后这个数每出现一次,对应索引的数就增加1,比如123234,那么新的数组的索引为1的值就加1,然后索引为2的值也加1,以此类推…
对于这个题,我们只需要把新数组每个元素的值除以2(得到个数)然后乘以索引(原来数的大小),但是对于个数,可能为奇数,或者偶数,偶数无须特殊处理,但是对于奇数,需要往后一个数借1,当然后面一个数的个数也需要更新
代码
class Solution {
public int arrayPairSum(int[] nums) {
int[] count = new int[20001];
for(int x:nums){
count[x + 10000]++;
}
// borrow是向上借一个
int ret = 0, borrow=0;
for(int i=-10000;i<=10000;i++){
// count[i]是总共出现了多少个, -borrow是i-1有没有借走一个
// +1是为了保证,count[i] - borrow是奇数的情况下,多算一个
ret += (count[i+10000] - borrow + 1) / 2 * i;
// 更新borrow,如果count[i] - borrow是奇数,就得从i+1借一个
// +2是java语言防止-1 % 2 == -1,因为每个计数的位置可能为0,而borrow可能为0,1,为了满足取模,所以加2
borrow = (count[i+10000] - borrow + 2) % 2;
}
return ret;
}
}