OJ刷题------分割等和子集（c++完整代码）

小鸽子。

于 2022-12-17 16:29:54 发布

阅读量165

点赞数

分类专栏： OJ刷题文章标签：算法动态规划 c++

本文链接：https://blog.csdn.net/qq_44258878/article/details/128354654

版权

OJ刷题专栏收录该内容

27 篇文章 4 订阅

订阅专栏

题目：

给定一个非空的正整数数组 nums(nums.length < 5) ，请判断能否将这些数字分成元素和相等的两部分。
输入：
1 5 11 5
输出：
true

分析

这道题可以换一种表述：给定一个只包含正整数的非空数组 nums[0]，判断是否可以从数组中选出一些数字，使得这些数字的和等于整个数组的元素和的一半。因此这个问题可以转换成「0-1 背包问题」。这道题与传统的「0-1背包问题」的区别在于，传统的「0-1 背包问题」要求选取的物品的重量之和不能超过背包的总容量，这道题则要求选取的数字的和恰好等于整个数组的元素和的一半。类似于传统的「0-1 背包问题」，可以使用动态规划求解。
具体可以看链接分割等和子集
代码参考链接分割等和子集代码

代码（C++完整代码）：

#include<iostream>
#include<vector>
#include<numeric>
using namespace std;
class Solution {
public:
	bool can(vector<int>& nums)
	{
		if (nums.size() == 0 || nums.size() == 1)//如果Nums数组的长度<2，那就不可能分成2半，所以可以直接返回false。
			return false;
		int sum = accumulate(nums.begin(), nums.end(), 0);
		if (sum % 2) return false;//如果数组之和是奇数，则不能分为两个和相等的部分。
		int n = nums.size();
		int target = sum / 2;//即target是总的和的一半。
		vector<bool> dp(target + 1, false);
		dp[0] = true;
		for (int i = 0; i < n; i++)
		{
			for (int j = target; j >= nums[i]; --j)
			{
				dp[j] = dp[j] || dp[j - nums[i]];
			}
		}
		return dp[target];
	}
};
int main() 
{
	Solution sol;
	vector<int> nums;
	int i;
	while (cin >> i)
	{
		nums.push_back(i);
		if (cin.get() == '\n')break;
	}
	int n;
	n = sol.can(nums);
	if (n == 0)cout << "false" << endl;
	else cout << "true" << endl;
	return 0;
}