一、题目介绍
给定长度为 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].
二、解题思路
(1)按照题目的要求,如果想得从1 到 n 的 min(ai, bi) 总和最大,那么需要将数组进行排序,然后取奇数位置上的数进行想加求和,即为结果。
(2)上一步利用贪心算法的思想,因为min(ai, bi)取两个数中的最小值作为结果,所以我们没有办法取到最大值。这样如果想保证总和最大,首先将数组从小到大排序,按照顺序分成n组,每组取最小数作为结果进行相加才能最大,由局部最优得到全局最优。
(3)我们可以分析一下,从第n组开始依次往前看,因为最大数无论和谁一组都无法取到,那只能退而求其次得到第二大的数,所以只能将它们放在一组。对于第n-1组而言,如果想获得第三大数,只能将它与第二大数放在一起,那这样的损失会更多,不仅不能取得最大值还无法取到第二大的值,因此为了能获得第二大数损失掉第三大数是值得的,所以将第三大数和第四大数放一起。这样依次类推,可以看出排序后再按顺序分组是正确的。
三、解题代码
第一种方法:使用sort排序
int arrayPairSum(vector<int>& nums) {
if(nums.empty())
return 0;
sort(nums.begin(), nums.end());
int l = 0;
int r = nums.size()-2;
int sum = 0;
while(l <= r)
{
if(l == r)
sum += nums[l];
else
sum += nums[l] + nums[r];
l += 2;
r -= 2;
}
return sum;
}
第二种方法:使用qsort排序,注意参数comp为一个全局函数不能是成员函数
int cmp(const void* a, const void* b)
{
return *(int*)a - *(int*)b; //从小到大排列
}
class Solution {
public:
int arrayPairSum(vector<int>& nums) {
if(nums.empty())
return 0;
qsort(&nums[0], nums.size(), sizeof(int), cmp);
int l = 0;
int r = nums.size()-2;
int sum = 0;
while(l <= r)
{
if(l == r)
sum += nums[l];
else
sum += nums[l] + nums[r];
l += 2;
r -= 2;
}
return sum;
}
};
四、解题结果
第一种方法结果:
第二种方法结果: