题目
给定一个包含 N 个正整数的集合,请你将它划分为两个集合 A1 和 A2,其中 A1 包含 n1 个元素,A2 包含 n2 个元素。
集合中可以包含相同元素。
用 S1 表示集合 A1 内所有元素之和,S2 表示集合 A2 内所有元素之和。
请你妥善划分,使得 |n1−n2| 尽可能小,并在此基础上 |S1−S2| 尽可能大。
输入格式
第一行包含整数 N。
第二行包含 N 个正整数。
输出格式
在一行中输出 |n1−n2| 和 |S1−S2|,两数之间空格隔开。
数据范围
2≤N≤105,
保证集合中各元素以及所有元素之和小于 231。
输入样例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
题解
题目首先要求两个集合元素个数差最小,所以尽量均分个数。
元素个数为偶数的时候,分开后,两个集合元素个数一样,差值为 0。
元素个数为奇数的时候,分开后,元素个数差值为 1。
然后要求子集合差值尽量大,那么一个集合里放小的元素,一个集合里放大的元素就行了。
n 为奇数的时候,不够均分,就在放大元素的集合里,多放一个元素。
所以:
将数组排序后,a[0] ~ a[n/2-1] 个放在一个集合里,a[n/2] ~ a[n-1 ]剩余的放在另一个集合里。
(因为除法结果是向下取整,所以整好可以这样分)
例如:
n = 5,就将数组排序,a[0] ~ a[1] 放在一个集合,a[2] ~ a[4]放在一个集合。然后分别对两个集合求和,小的和为 s1 ,大的和为 s2。 输出:1 s2 - s1。
n = 6,就将数组排序,a[0] ~ a[2] 放在一个集合,a[3] ~ a[6] 放在一个集合。然后分别对两个集合求和,小的和为 s1 ,大的和为 s2。 输出:0 s2 - s1。
#include <iostream>
#include <algorithm>
using namespace std;
const int N = 100010;
int a[N];
int sum(int a[], int l, int r)//求数组 a[l] ~ a[r] 的和
{
int res = 0;
for(int i = l; i <= r; i++)
res += a[i];
return res;
}
int main()
{
int n;
cin >> n;
for(int i = 0; i < n; i++)
{
cin >> a[i];
}
sort(a,a + n);//排序
int s1, s2;
s1 = sum(a, 0, n/2-1);//s1 的和小
s2 = sum(a, n/2, n - 1);//s2 的和大
cout << n % 2 << " " << s2 - s1;
}