我的PAT-ADVANCED代码仓:https://github.com/617076674/PAT-ADVANCED
原题链接:https://pintia.cn/problem-sets/994805342720868352/problems/994805357258326016
题目描述:
题目翻译:
1113 整数集分区
给定一个集合,其中有N(> 1)个正整数,你需要将其分成A1和A2两个部分,A1中有n1个数字,A2中有n2个数字。S1、S2分半表示A1、A2中的数字和。你需要选取一种分法,使得|n1 - n2|最小且|S1 - S2|最大。
输入格式:
每个输入文件包含一个测试用例。对每个测试用例,第一行给出一个整数N(2 < N <= 10 ^ 5),下一行紧跟着N个正整数,每个数字由一个空格分隔。题目保证所有的数字及其和均小于2 ^ 31。
输出格式:
对每个测试用例,在一行中输出|n1 - n2|和|S1 - S2|,由一个空格分隔。
输入样例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
知识点:贪心算法
思路:对数组排序后取一半一半划分即可
时间复杂度是O(NlogN)。空间复杂度是O(N)。
C++代码:
#include<iostream>
#include<algorithm>
using namespace std;
int main() {
int N;
scanf("%d", &N);
int nums[N];
for(int i = 0; i < N; i++) {
scanf("%d", &nums[i]);
}
sort(nums, nums + N);
int sum1 = 0, sum2 = 0;
if(N % 2 == 0) {
printf("0 ");
} else {
printf("1 ");
}
for(int i = 0; i < N / 2; i++) {
sum1 += nums[i];
}
for(int i = N / 2; i < N; i++) {
sum2 += nums[i];
}
printf("%d\n", sum2 - sum1);
return 0;
}
C++解题报告: