561. Array Partition I (数组分区 I)
1. 题目翻译
给定一个大小为2n的整数数组,你需要将它分成n组,每组两个数字(a1, b1), (a2, b2), …, (an, bn)。分组的依据是,取每组最小的数共n个,使他们的和尽可能大。
例子:
输入: [1,4,3,2]
输出: 4
解释: n为2, 最大和为4 = min(1, 2) + min(3, 4).
备注:
- n是正整数,范围为 [1, 10000].
- 所有数组中的整数的范围为 [-10000, 10000].
2. 解题方法
1,2,3,4 在这4个数中,4是不可能得到的,因为没有别的数比它更大。想要得到3必须将3与4分在一组。剩下的1和2只能取1。这道题其实就是取最大数与次大数,然后将这个数从原数组去掉,继续去最大数与次大数。最后将所有次大数加在一起。所以可以将原数组排序,然后取奇数位的所有值的和。
3. 代码
//Runtime: 75ms
class Solution {
public:
int arrayPairSum(vector<int>& nums) {
sort(nums.begin(),nums.end());
int res = 0;
for(int i = 0;i<nums.size();i+=2)
res += nums[i];
return res;
}
};
如果不想使用自带的sort方法,可以使用下面自己写的快速排序的代码。
//Runtime: 78ms
class Solution {
private:
int partition(vector<int>& nums, int left, int right){
int i =left+1,j=i;
for(i;j<=right;j++){
if(nums[j]<=nums[left]){
int temp = nums[j];
nums[j] = nums[i];
nums[i] = temp;
i++;
}
}
int temp = nums[i-1];
nums[i-1] = nums[left];
nums[left] = temp;
return i-1;
}
void quicksort(vector<int>& nums, int left, int right){
if(left<right){
int index = partition(nums,left,right);
quicksort(nums,left,index-1);
quicksort(nums,index+1,right);
}
}
public:
int arrayPairSum(vector<int>& nums) {
quicksort(nums,0,nums.size()-1);
int res = 0;
for(int i = 0;i<nums.size();i+=2)
res += nums[i];
return res;
}
};