题目来源:https://leetcode.cn/problems/NUPfPr/
大致题意:
给定一个数组,求数组能否分成两个和相等的子序列
思路
相当于求数组是否有子序列和可以等于数组和的一半
于是可以先求出数组和,判断数组和是否是偶数,若不是直接返回 false
若是,可以使用动态规划解决
使用二维布尔数组 dp[i][j] 表示数组前 i 个元素是否可以组成和为 j 的序列
那么有
dp[i][j] = dp[i - 1][j] | dp[i - 1][j - nums[i]]
初始时,dp[i][0] = true
因为 dp[i] 的值只与 dp[i - 1] 有关,且 dp[i][j] 只与 dp[i - 1][k] (其中 k 小于等于 j)有关,所以可以用滚动数组优化
dp[j] = dp[j] | dp[j - nums[i]
遍历时,需要倒序进行遍历,防止本轮更新的较小位置元素值影响到后面元素值
初始时,dp[0] = true
代码:
public boolean canPartition(int[] nums) {
int sum = Arrays.stream(nums).sum();
if (sum % 2 != 0) {
return false;
}
int n = nums.length;
int t = sum / 2;
// 滚动数组
boolean[] dp = new boolean[t + 1];
// 初始化
dp[0] = true;
for (int num : nums) {
for (int j = t; j >= num; j--) {
// 状态转移方程
dp[j] = dp[j] || dp[j - num];
}
// 只要当前子序列可以组成 t,直接返回结果
if (dp[t]) {
return true;
}
}
return false;
}