力扣刷题之561. 数组拆分 I
作者:互联网
题目简述
给定长度为 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 <= 10^4
- nums.length == 2 * n
- -10^4 <= nums[i] <= 10^4
题目分析
题目看似非常难理解,但其实根据给的例子稍微想一下就能明白,这道题实际上就是把数组的数字排序后相邻两个元素组队即可,有点像田忌赛马,如果把越小的数和越大的数结合,那么最后的结果就会越偏小,所以只有相邻的元素相结合才能使结果最大。代码如下:
class Solution {
public:
int arrayPairSum(vector<int>& nums) {
int n=nums.size();
int sum=0;
sort(nums.begin(),nums.end());
for(int i=0;i<n;i++){
sum+=min(nums[i],nums[i+1]);
i++;
}
return sum;
}
};
当然,如果更善于发现规律的话就会发现实际上排序后取两元素的最小值会把下标为奇数的元素留下来,所以只用把所有下标为奇数的元素加起来即可,这样耗时会更少。代码如下:
class Solution {
public:
int arrayPairSum(vector<int>& nums) {
int n=nums.size();
int sum=0;
sort(nums.begin(),nums.end());
for(int i=0;i<n;i+=2){
sum+=nums[i];
}
return sum;
}
};
标签:元素,nums,int,561,sum,min,力扣,数组,刷题 来源: https://blog.csdn.net/mr_fangf/article/details/113822577