LeetCode.561 数组拆分 I

这是我参与11月更文挑战的第27天,活动详情查看:2021最后一次更文挑战

题目描述:

561. 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)

给定长度为 2n 的整数数组 nums ,你的任务是将这些数分成 n 对, 例如 (a1, b1), (a2, b2), ..., (an, bn),使得从 1 到 nmin(ai, bi) 总和最大。

返回该 最大总和 。

示例一

输入: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
复制代码

示例二

输入: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

思路分析

贪心

假设排完序的结果为a1<=b1<=a2<=b2<=...<=an<=bn

那么a1应该跟谁一组呢?

a1作为全局最小值,无论跟谁一组a1都会被累加进答案,相反,a1的搭档会被永久排除。

既然如此,莫不如排除一个较小的数,即给a1找一个“最小的搭档”b1。

当a1、b1被处理之后,a2同理分析。

所以,最终选择a1,a2,...,an会得到最好的结果。

这是按贪心的思路来的,然后官方答案中还有此解法的另一种解读,顺带推理过程的,感兴趣的去看看。

AC代码

class Solution {
    fun arrayPairSum(nums: IntArray): Int {
        nums.sort()
        val len = nums.size
        var maxNum = 0
        for (index in 0 until len step 2){
             maxNum += nums[index]
        }   
        return maxNum
    }
}
复制代码

总结

同一套代码,有不同的解题思路,贪心算法和数学证明法,殊途同归。

参考

数组拆分 I - 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)

【宫水三叶の相信科学系列】反证法证明贪心算法的正确性 - 数组拆分 I - 力扣(LeetCode) (leetcode-cn.com)

おすすめ

転載: juejin.im/post/7035604977632411655