题目

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.

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

思路一

要把整个数组分为一对一对的,然后每对取最小的那个,使最小的加起来值最大。可以想到,小的一定要跟小的配对,因为最小的那个必然要算进去,如果跟它配对的那个比较大,那这个大的就浪费了。所以,最佳的状态就是,整个数组从小到大排好序,然后我们取奇数项相加,得到的和就是结果。时间复杂度是O(nlogn)。

思路一代码

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

思路二

思路一的代码提交上去可以Accept,但是效率太低。原来还有另外一种方法,可以在近似O(n)时间内解决这个问题。

这个方法叫做桶排序(bucket sort)。这其实是一种浪费空间换取时间的方法,并且需要在特定条件下才有用。假设给定一个数组[1, 4, 3, 2],根据题目要求,数组最大长度是20000, 每个元素的值最小为-10000, 最大为10000,那么我们就可以开一个长度为20000的数组,初始化为0,然后遍历原数组,原数组的值作为新数组的下标,有就加1,然后我们在新数组从头开始遍历,就能直接输出“排好序”的数组了。既然能得到一个排好序的数组,跟思路一一样,直接就能得出答案。

思路二代码

    int arrayPairSumBucketSort(vector<int>& nums) {
      vector<int> bucket(10000 * 2 + 1, 0);
      int lower = 10000 * 2;
      for (int i = 0; i < nums.size(); ++i) {
        bucket[nums[i] + 10000] += 1;
        if (nums[i] + 10000 < lower) {
          lower = nums[i] + 10000;
        }
      }

      int count = 0;
      int sum = 0;
      bool skip = false;
      while (count < nums.size() / 2) {
        if (bucket[lower] > 0 && !skip) {
          sum += lower - 10000;
          skip = true;
          ++count;
          bucket[lower]--;
        } else if (bucket[lower] > 0 && skip) {
          skip = false;
          bucket[lower]--;
        } else {
          ++lower;
        }
      }
      return sum;
    }