C/C++编程题刷题：分组

分组

【题目描述】2020年一共有n名员工参加入职培训，第i名员工拥有一个战力值x[i]。
在入职培训中有一个小活动需要把所有员工分为A、B两组进行比拼对决。
为了让活动更有趣味性，小Q希望两组的人数尽量均分(即两组人数之差不超过1)，并且要让两组尽量势均力敌(A组的战力值之和与B组的战力值之和的差值尽量小)。
小Q把这个问题交给你了，希望你帮他完成这个分组问题。

输入描述

输入的第一行包含一个正整数T(1<=T<=10),表示测试用例数。
对于每一组测试用例，输入的一行包括一个正整数n(2<=n<=100),表示员工的数量。

接下来的一行,包含n个正整数x[i](1<=x[i]<=105）

保证所有员工的战力值之和小于等于105。

输出描述

输出两个整数, 从小到大依次输出两组的战力值之和。

示例1

输入

2 4 5 9 4 7 8 9 13 18 10 12 4 18 3

输出

12 13 43 44

#include<iostream>
#include<string>
#include<vector>
#include<algorithm>
using namespace std;
void solution(vector<int> &nums) {
	int n = nums.size();
	int sum = 0;
	int cap = 0;
	for (auto x : nums) sum += x;
	//计算背包容量
	if (sum % 2 == 1)
		cap = (sum + 1) >> 1;
	else
		cap = sum >> 1;
	//计算一组最多需要多少人
	int cnt = (n % 2 == 1) ? ((n + 1) >> 1) : (n >> 1);
	//对于前i个物品，当背包容量为j时，若DP[i][j]为true,则说明背包可以装得下
	vector<vector<bool>> dp(n + 1, vector<bool>(cap + 1, false)); //
	//base case 初始条件
	//当背包数目为0的时候 就相当于装满了 
	for (int i = 0; i <= n; i++)
		dp[i][0] = true; 
	//当没有物品可以选择的时候  肯定没有办法装满背包
	for (int i = 1; i <= cap; i++)
		dp[0][i] = false;
	//状态转移
	for (int i = 1; i <= n; i++)
		for (int j = cap; j >= nums[i - 1]; --j) {
			for (int k = cnt; k >= 1; --k)
			{
				dp[k][j] = dp[k][j] | dp[k - 1][j - nums[i - 1]];
			}
		}
	
	int t = 0;
	/*从cnt行开始找到最大的背包容量j*/
	for (int j = cap; j >= 0; --j) {
		if (dp[cnt][j]) {
			t = j;
			break;
		}
	}
	cout << min(t, sum - t) << " " << max(t, sum - t) << endl;
}
int main() {
	int t;
	cin >> t;
	while (t--) {
		int n;
		cin >> n;
		vector<int> nums(n);
		for (int i = 0; i < n; i++) {
			cin >> nums[i];
		}
		solution(nums);
	}
	system("pause");
	return 0;

}

参考解法：https://blog.csdn.net/u014106644/article/details/83959752?utm_medium=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.nonecase&depth_1-utm_source=distribute.pc_relevant.none-task-blog-BlogCommendFromMachineLearnPai2-2.nonecase