一起养成写作习惯!这是我参与「掘金日新计划 · 4 月更文挑战」的第9天,点击查看活动详情。
一 描述
561. 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)
给定长度为 2n 的整数数组 nums ,你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), ..., (an, bn) ,使得从 1 到 n 的 min(ai, bi) 总和最大。
返回该 最大总和 。
示例 1:
输入:nums = [1,4,3,2]
输出:4
解释:所有可能的分法(忽略元素顺序)为:
1. (1, 4), (2, 3) -> min(1, 4) + min(2, 3) = 1 + 2 = 3
2. (1, 3), (2, 4) -> min(1, 3) + min(2, 4) = 1 + 2 = 3
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 <= 10^4
- nums.length == 2 * n
- -10^4 <= nums[i] <= 10^4
二 分析
- 根据题目要求,一堆min的和最大,则可推得每个min应该尽可能的大,要让每个min尽可能的大,则每个min中的两个数应该尽可能接近或者相等。
- 故基于此思路,先将数组从小到大排序,然后从前往后两两成对即可满足要求。
三 答案
class Solution {
public:
int min(int a, int b){
return a<b ? a:b;
}
int arrayPairSum(vector<int>& nums) {
int sum = 0;
sort(nums.begin(), nums.end());
for(int i = 0; i < nums.size(); i += 2){
sum += min(nums[i], nums[i+1]);
}
return sum;
}
};
四 总结
其实还可以简化下,排序后,如果不要求成对,实际上可以不进行min这一步的比较,从第一项开始,隔位取值再相加就是最终的答案。
参考
C++ 手写快排:比官方sort要慢不少呀.. - 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)