给定一个包含 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
思路
题目上说使得|n1-n2|尽可能小,所以只能是0或者1。在此基础上有要求|s1-s2|最大,所以就是最大的n个数与最小的m个数的差值
#include <iostream>
#include <algorithm>
using namespace std;
int arr[100005];
int sum[100005];
int main(void)
{
int n;
cin >> n;
for (int i = 1; i <= n; i++)
{
cin >> arr[i];
}
sort(arr+1, arr + n + 1);//排序
for (int i = 1; i <= n; i++)
{
sum[i] = sum[i - 1] + arr[i];//利用前缀和方便求出两个区间的和
}
//如果为奇数,就划分不出两个数量一样的区间,输出1。答案为后n-n+1个数的和与前n-1个数的和的差值
if (n % 2)cout << 1 << ' ' << sum[n] - 2*sum[(n - 1) / 2];
//如果为偶数,可以划分出两个数量一样的区间,输出0。答案为后n-/2个数的和与前n、2个数的和的差值
else cout << 0 << ' ' << sum[n] - 2*sum[n / 2];
return 0;
}