【Leetcode】416. 分割等和子集

题目:416

题意:给定一个只包含正整数的非空数组。问是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。

题解:动态规划

dp[i][j]表示从数组的 [0, i] 这个子区间内挑选一些正整数,每个数只能用一次,使得这些数的和恰好等于j

状态转移方程

 dp[i][j] = \left\{\begin{matrix} dp[i-1][j]& \\ true &if j == nums[i] \\ dp[i-1][j-nums[i]] || dp[i-1][j] & if j > nums[i] \end{matrix}\right.

代码:

class Solution {
public:
	bool canPartition(vector<int>& nums) {
		int sum = 0;
		int len = nums.size();

		//if (len == 0)return false;  //题目给出数组非空
		for (int i = 0; i < len; ++i)
			sum += nums[i];

		if (sum % 2 == 1)return false;  //特判

		sum = sum >> 1;
		//dp[i][j]表示从数组的 [0, i] 这个子区间内挑选一些正整数,每个数只能用一次,使得这些数的和恰好等于j
		vector<vector<int> > dp = vector<vector<int> >(len, vector<int>(sum + 1, 0));
		//边界
		if (nums[0] <= sum) dp[0][nums[0]] = true; 

		//状态转移方程
		for(int i = 1; i < len; ++i)
			for (int j = 0; j <= sum; ++j) {
				dp[i][j] = dp[i - 1][j];

				if (nums[i] == j)dp[i][j] = true;
				if (nums[i] < j)
					dp[i][j] = dp[i][j] || dp[i - 1][j - nums[i]];
			}
		return dp[len - 1][sum];
	}
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
题目描述: 给定一个只包含正整数的非空数组,是否可以将这个数组分成两个子集,使得两个子集的元素和相等。 示例: 输入:[1, 5, 11, 5] 输出:true 解释:数组可以分割成 [1, 5, 5] 和 [11]。 解题思路: 这是一道经典的 0-1 背包问题,可以使用动态规划或者回溯算法解决。 使用回溯算法,需要定义一个 backtrack 函数,该函数有三个参数: - 数组 nums; - 当前处理到的数组下标 index; - 当前已经选择的元素和 leftSum。 回溯过程中,如果 leftSum 等于数组元素和的一半,那么就可以直接返回 true。如果 leftSum 大于数组元素和的一半,那么就可以直接返回 false。如果 index 到达数组末尾,那么就可以直接返回 false。 否则,就对于当前元素,有选择和不选择两种情况。如果选择当前元素,那么 leftSum 就加上当前元素的值,index 就加 1。如果不选择当前元素,那么 leftSum 不变,index 也加 1。最终返回所有可能性的结果中是否有 true。 Java 代码实现: class Solution { public boolean canPartition(int[] nums) { int sum = 0; for (int num : nums) { sum += num; } if (sum % 2 != 0) { return false; } Arrays.sort(nums); return backtrack(nums, nums.length - 1, sum / 2); } private boolean backtrack(int[] nums, int index, int leftSum) { if (leftSum == 0) { return true; } if (leftSum < 0 || index < 0 || leftSum < nums[index]) { return false; } return backtrack(nums, index - 1, leftSum - nums[index]) || backtrack(nums, index - 1, leftSum); } }
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值