1.题目描述:
给你一个只包含正整数的非空数组nums。请你判断是否可以将这个数组分割成两个子集,使得两个子集的元素和相等。
2.动态规划:
首先若能分割数组元素之和必须是偶数,也就是判断元素之和是否有总和/2,数组排序一下感觉回溯也可以写。动态规划转化为01背包。背包问题建议手动填写dp数组验证。细节见代码注释:
二维dp数组:
class Solution {
public boolean canPartition(int[] nums) {
int sum = 0;
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
}
if (sum % 2 == 1) return false;
int bpSize = sum / 2;
int gSize = nums.length;
int[][] dp = new int[gSize][bpSize + 1];//增长1列
for (int i = 0; i < bpSize + 1; i++) {//初始化二维数组的首行(i-1)的需求
if (i >= nums[0]) dp[0][i] = nums[0];
}
for (int i = 1; i < gSize; i++) {//先遍历物品,再遍历背包,先行后列
for (int j = 0; j < bpSize + 1; j++) {
if (j < nums[i]) dp[i][j] = dp[i - 1][j];
else dp[i][j] = Math.max(dp[i - 1][j], dp[i - 1][j - nums[i]] + nums[i]);
//状态转移方程,dp[i][j]的推导分成两种情况:装入当前i物品(扣除i的价值能装的最大价值+i的价值),不装当前i物品(延续上一行同列的值)
}
}
return dp[gSize - 1][bpSize] == bpSize;
}
}
一维dp数组:
class Solution {
public boolean canPartition(int[] nums) {
int sum = 0;
for (int i = 0; i < nums.length; i++) {
sum += nums[i];
}
if (sum % 2 == 1) return false;
int bpSize = sum / 2;
int gSize = nums.length;
int[] dp = new int[bpSize + 1];//增长1列
for (int i = 0; i < gSize; i++) {//依旧是先遍历物品,再遍历背包,先行后列
for (int j = bpSize; j >= nums[i]; j--) {//这里j < nums[i]的情况直接记录上层的dp[j]
dp[j] = Math.max(dp[j], dp[j - nums[i]] + nums[i]);
//逆序遍历,否则顺序遍历时dp[j - nums[i]]将是本层的数据而非上一层的数据,这里首行初始化都放在一起了
}
}
return dp[bpSize] == bpSize;
}
}