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;
}