给定长度为 2n 的整数数组 nums ,你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), …, (an, bn) ,使得从 1 到 n 的 min(ai, bi) 总和最大。
返回该 最大总和 。
示例 1:
输入:nums = [1,4,3,2]
输出:4
解释:所有可能的分法(忽略元素顺序)为:
- (1, 4), (2, 3) -> min(1, 4) + min(2, 3) = 1 + 2 = 3
- (1, 3), (2, 4) -> min(1, 3) + min(2, 4) = 1 + 2 = 3
- (1, 2), (3, 4) -> min(1, 2) + min(3, 4) = 1 + 3 = 4
所以最大总和为 4
示例 2:
输入:nums = [6,2,6,5,1,2]
输出:9
解释:最优的分法为 (2, 1), (2, 5), (6, 6). min(2, 1) + min(2, 5) + min(6, 6) = 1 + 2 + 6 = 9
提示:
1 <= n <= 104
nums.length == 2 * n
-104 <= nums[i] <= 104
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/array-partition-i
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
两种解法:
1.先对数组从小到大排序(调用sort函数),然后把偶数下标的值加起来返回;
2. 由于题目限制了数组值的范围为-10000~10000;所以可以采用一个临时数组来记录每个值出现的次数,相当于完成排序。然后遍历临时数组,根据index值取出index为偶数的值相加。这里提供了除了sort外的另外一种排序思路。
为什么是偶数下标的值相加就能得出最大的总和,leetcode官方给出了正向的数学推导,也有别的同学给出了反向证明法(贪心算法,现在还不是很清楚为什么是贪心算法,先mark一下)。
官方给的正向推导很好理解,就是一道证明题,在这里贴出原文网址:https://leetcode.cn/problems/array-partition-i/solution/shu-zu-chai-fen-i-by-leetcode-solution-9m9y/
有兴趣的可以学习一下,大概意思就是取任意(ai,bi),能得到有n - i个数大于ai,n - i个数大于bi,ai < bi,也就是「从小到大」至多排在第 2n - 2(n-i+1) + 1 = 2(i-1) + 1个位置 然后就有排过序的数组c,ai=C2(i−1)+1。 比如a2 = c3(推导过程中i是从1开始),所以最后的和就是c1+c3+c5…
class Solution {
public:
int arrayPairSum(vector<int>& nums) {
#if 0
sort(nums.begin(), nums.end());
int sum = 0;
for(int i = 0; i < nums.size(); i += 2)
{
sum += nums[i];
}
return sum;
}
#else
int tmp_vec[20001] = {0};
for (int i = 0; i < nums.size(); i++)
{
tmp_vec[nums[i] + 10000]++;
}
int sum = 0, idx = 0;
for (int i = 0; i < 20001;)
{
if (tmp_vec[i] != 0)
{
if ((idx % 2) == 0)
{
sum += i - 10000;
}
tmp_vec[i]--;
idx++;
}
else
{
i++;
}
}
return sum;
}
#endif
};