Leetcode题解-561. Array Partition I

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;
    }