刷题-力扣-461. 数组拆分 I

561. 数组拆分 I

题目链接

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/array-partition-i/
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

题目描述

给定长度为 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 <= 104
  • nums.length == 2 * n
  • -104 <= nums[i] <= 104

题目分析

  1. 根据题目描述选取nums中组合最大总和
  2. 使用快速排序nums,nums偶数下标的和即为所求

代码

class Solution {
public:
    int arrayPairSum(vector<int>& nums) {
        quickSort(nums, 0, nums.size() - 1);
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (i % 2 == 0) sum += nums[i];
        }
        return sum;
    }
private:
    void quickSort(vector<int>& nums, int l, int r) {
        if (l >= r) return;
        int i = l;
        int j = r;
        int x = nums[l];        // 基准数
        while (i < j) {
            while (i < j && x <= nums[j]) {
                j--;
            }
            if (i < j) {
                nums[i++] = nums[j];
            }
            while (i < j && x > nums[i]) {
                i++;
            }
            if (i < j) {
                nums[j--] = nums[i];
            }
        }
        nums[i] = x;
        quickSort(nums, l, i - 1);
        quickSort(nums, i + 1, r);
    }
};
原文地址:https://www.cnblogs.com/HanYG/p/14407433.html