AcWing 1603.整数集合划分
给定一个包含 NN 个正整数的集合,请你将它划分为两个集合 A1A1 和 A2A2,其中 A1A1 包含 n1n1 个元素,A2A2 包含 n2n2 个元素。
集合中可以包含相同元素。
用 S1S1 表示集合 A1A1 内所有元素之和,S2S2 表示集合 A2A2 内所有元素之和。
请你妥善划分,使得 |n1−n2||n1−n2| 尽可能小,并在此基础上 |S1−S2||S1−S2| 尽可能大。
输入格式
第一行包含整数 NN。
第二行包含 NN 个正整数。
输出格式
再一行中输出 |n1−n2||n1−n2| 和 |S1−S2||S1−S2|,两数之间空格隔开。
数据范围
2≤N≤1052≤N≤105,
保证集合中各元素以及所有元素之和小于 231231。
输入样例1:
10
23 8 10 99 46 2333 46 1 666 555
输出样例1:
0 3611
输入样例2:
13
110 79 218 69 3721 100 29 135 2 6 13 5188 85
输出样例2:
1 9359
思路:
当n为偶数时,|n1 - n2| = 0,当n为奇数时|n1 - n2| = 1. 而且前一个集合一定比后一个集合少一个元素,即n1<n2. 我们只需要把所有相对小的元素放到第一个集合,把所有相对大的元素放到第二个集合,就可以使|S1 - S2| 最大。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int n;
int a[N];
int main()
{
cin >> n;
for(int i = 0; i < n; i ++ )
scanf("%d",&a[i]);
sort(a,a + n);
int s1 = 0, s2 = 0;
for(int i = 0; i < n/2; i ++ ) s1 += a[i];
for(int j = n/2; j < n; j ++ ) s2 += a[j];
printf("%d %d",n % 2,s2 - s1);
return 0;
}