LeetCode 561. 数组拆分 I

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

返回该 最大总和 。

在长度为2n时,min(ai, bi)最大时,ai、bi必须是两个最大的数,此时min(ai, bi)为数组中第二大的数,之后去掉最大的两个数还剩2n-2个数,递归计算即可。方法是将数组排序,之后取偶数位的和:

	void quickSort(vector<int> &nums, size_t l, size_t r) {
        if (l >= r) {
            return;
        }

        default_random_engine e;
        uniform_int_distribution<size_t> u(l, r);
        size_t randIndex = u(e);
        swap(nums[randIndex], nums[r]);
        size_t small = l, cur = l;

        while (cur < r) {
            if (nums[cur] < nums[r]) {
                swap(nums[cur], nums[small]);
                ++small;
            } 
            ++cur;
        }
        swap(nums[small], nums[r]);

        if (small != l) {
            quickSort(nums, l, small - 1);
        }
        if (small != r) {
            quickSort(nums, small + 1, r);
        }
    }

    int arrayPairSum(vector<int>& nums) {
        size_t sz = nums.size();
        quickSort(nums, 0, sz - 1);

        int sum = 0;
        for (size_t i = 0; i < sz; i += 2) {
            sum += nums[i];
        }
        return sum;
    }
posted @ 2021-03-13 22:32  epiphanyy  阅读(12)  评论(0)    收藏  举报  来源