[LeetCode] 561. Array Partition I

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).

备注：

1. n是正整数，范围为 [1, 10000].
2. 所有数组中的整数的范围为 [-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;
    }
};