LeetCode 561:数组拆分 I

LeetCode 561:数组拆分 I

题目描述

给定长度为 2n 的数组, 你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), …, (an, bn) ，使得从1 到 n 的 min(ai, bi) 总和最大。

【示例 1】

输入: [1,4,3,2]

输出: 4
解释: n 等于 2, 最大总和为 4 = min(1, 2) + min(3, 4).

提示:

n 是正整数,范围在 [1, 10000].
数组中的元素范围在 [-10000, 10000].

解题

    假定$min(a_i,b_i)总是为a_i$，为了使得$\sum min(a_i,b_i)$最大，只需要保证$a_{i+1}\ge b_i$。也就是说，为了使得最小值之和最大，只有将较小的树组成一对，较大的数组成一对，这样较大的数才能参与到求和运算中，使得和的值最大。

排序

    为了组队，先将整个数组排序，将相邻两个数字组合就可以划分成n对。

class Solution {
public:
    int arrayPairSum(vector<int>& nums) {
        int sum=0;
        sort(nums.begin(), nums.end());
        for (int i=0; i<nums.size(); i+=2)
            sum += nums[i];
        return sum;
    }
};

借助直方图

    直方图可以统计各个数值出现的数目，题目已经给出数值的范围，因此可以构造一个-10000~10000的直方图统计。直方图统计把数组的位置结构打散，我们得到的就是一组[-10000,10000]范围内的数值，且有每个数值的数目。比如数值1有50个，那么这些数字配对的话，可以组成25对，每个最小是为1，这部分的和为50/21=25。由此借助直方图就可以计算和值。
    但是会有一个问题，那就是每个数值的数目不一定是偶数，上一个数值的数目为奇数时会影响下一个数值部分的和值计算。显然地，上一个数值配对后只能剩下一个数或者恰好配对。上一个数值剩下的那一个数将参与到当前数值的配对中，当前数值会优先分配一个数字与之配对，剩下的n-1个数字再继续配对。因此需要一个变量来表示上一个数值配对后剩下的数值个数（0或1），在当前数值配对后对这个变量进行更新。另外一点就是，假如当前配对剩下一个数值，这个数值在参与配对后，与下一个数值相比，肯定是最小值，所以一定会参与到求和计算中，因此在当前部分就可以直接并入的和值当中。
    假如上一个数值配对后剩下的数值数目用flag表示，flag=0或1，用hist表示直方图数组，i表示当前位置，那么“
    hist[i]表示当前数值的数目，(hist[i]-flag)表示当前可以配对的数目，假如它为奇数，那(hist[i]-flag)/2i配对之后最小值的和，但是还剩下一个未配对的数值，且已经说明了，这个数值一定会参与求和运算，因此(hist[i]-flag+1)/2就能够保证当前数值参与到求和运算中，在下一个数值求和时也只需要对下一个数值部分求和即可。此外，(hist[i]-flag)为偶数时(hist[i]-flag+1)/2对结果并没有影响。

class Solution {
public:
    int arrayPairSum(vector<int>& nums) {
        vector<int> hist(20001, 0);
        for (int i=0; i<nums.size(); ++i)
            hist[nums[i]+10000] += 1;
        
        int sum = 0;
        int flag = 0;
        for (int i=-10000; i<=10000; ++i){
            sum += (hist[i+10000]-flag+1)/2*i;  //+1是保证多出的一个数值参与求和
            flag = (hist[i+10000]-flag) & 1;    //更新flag
        }
        
        return sum;
    }
};