日新刷题 - 561. 数组拆分 I

111 阅读1分钟

一起养成写作习惯!这是我参与「掘金日新计划 · 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

二 分析

  1. 根据题目要求,一堆min的和最大,则可推得每个min应该尽可能的大,要让每个min尽可能的大,则每个min中的两个数应该尽可能接近或者相等。
  2. 故基于此思路,先将数组从小到大排序,然后从前往后两两成对即可满足要求。

三 答案

 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)

Javascript 田忌赛马 - 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)

C 双指针 - 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)