Leetcode题解-561. Array Partition I
Given an array of 2n integers, your task is to group these integers into n pairs of integer, say (a1, b1), (a2, b2), …, (an, bn) which makes sum of min(ai, bi) for all i from 1 to n as large as possible.
Example 1:
Input: [1,4,3,2]
Output: 4
Explanation: n is 2, and the maximum sum of pairs is 4 = min(1, 2) + min(3, 4).
Note:
- n is a positive integer, which is in the range of [1, 10000]. All the
- integers in the array will be in the range of [-10000, 10000].
思路
可以简单地想到,每次取数组中第一大和第二大的整数组合成pair,然后将第二大的整数存起来,把第一大和第二大的整数从数组中去掉,如此往复直到数组中没有数,然后把这些存起来的第二大的整数相加,这样就能保证最后的和是最大的。将这种思路换另外的思路来实现,就是将数组从小到大排序,从1开始计数,取计数为奇数的整数相加就是所需要的结果。
因为题目已知数组的大小范围以及整数的范围,所以可以利用桶排算法,开一个足够大的数组用数组下标来代表桶(一个数值一个桶),将O(nlogn)的复杂度降低到O(n)。
代码
小细节:因为数的范围是 [-10000, 10000],所以将所有整数+10000使得能够对应上新建数组下标;只有当桶内整数个数为0(数组一个位置为0)时才进入下一个桶;取数规律为:取-不取-取-不取……
int arrayPairSum(vector<int>& nums) {
vector<int> result(20001,0);
int sum = 0, l = nums.size();
for(int i = 0; i < l; i++){
result[nums[i]+10000]++;
}
bool flag = true;
for(int i = 0; i < 20001; ){
if(flag && result[i]){
sum += i-10000;
result[i]--;
flag = false;
}
else if((!flag) && result[i]){
result[i]--;
flag = true;
}
else{
i++;
}
}
return sum;
}