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

题目:

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

分析

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

代码(C++完整代码):

#include
#include
#include
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;
}

运行结果

OJ刷题------分割等和子集(c++完整代码)_第1张图片

你可能感兴趣的:(OJ刷题,算法,动态规划,c++)