Leetcode题解-561. Array Partition I

来源:互联网 发布:苹果7蜂窝移动数据设置 编辑:程序博客网 时间:2024/06/13 14:24

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;    }
原创粉丝点击