旅游做网站优化王
题目来源:https://leetcode.cn/problems/target-sum/description/
C++题解(来源代码随想录):将该问题转为01背包问题。
假设加法的总和为x,那么减法对应的总和就是sum - x。所以我们要求的是 x - (sum - x) = target。x = (target + sum) / 2。此时问题就转化为,装满容量为x的背包,有几种方法。
- 确定dp数组以及下标的含义。(1)dp[j] 表示:填满j(包括j)这么大容积的包,有dp[j]种方法。(2)也可以使用二维dp数组来求解本题,dp[i][j]:使用 下标为[0, i]的nums[i]能够凑满j(包括j)这么大容量的包,有dp[i][j]种方法。
- 确定递推公式。只要搞到nums[i],凑成dp[j]就有dp[j - nums[i]] 种方法。
- dp数组如何初始化。dp[0] = 1。
- 确定遍历顺序。对于01背包问题一维dp的遍历,nums放在外循环,target在内循环,且内循环倒序。
// 代码随想录版本
class Solution {
public:int findTargetSumWays(vector<int>& nums, int S) {int sum = 0;for (int i = 0; i < nums.size(); i++) sum += nums[i];if (abs(S) > sum) return 0; // 此时没有方案if ((S + sum) % 2 == 1) return 0; // 此时没有方案int bagSize = (S + sum) / 2;vector<int> dp(bagSize + 1, 0);dp[0] = 1;for (int i = 0; i < nums.size(); i++) {for (int j = bagSize; j >= nums[i]; j--) {dp[j] += dp[j - nums[i]];}}return dp[bagSize];}
};
// 一维数组版本
class Solution {
public:int findTargetSumWays(vector<int>& nums, int target) {// left + right = sum;// left - right = target;// left = (sum + target) / 2;// 01背包:背包总量left,价值和为j的个数为dp[j]// dp[j] = dp[j]+dp[j-nums[i]]int len = nums.size();int sum = 0;for(int i = 0; i < len; i++) {sum = sum + nums[i];}if(sum < target || target < -sum) return 0;else if((sum - target) % 2 == 1) return 0;int left = (sum + target) / 2;vector<int> dp(left+1, 0);dp[0] = 1; // 初始化if(nums[0] <= left) dp[nums[0]]++; // 考虑left = nums[0] = 0的情况for(int i = 1; i < len; i++) {for(int j = left; j >= nums[i]; j--) {dp[j] = dp[j] + dp[j-nums[i]];}}return dp[left];}
};
// 二维数组版本
class Solution {
public:int findTargetSumWays(vector<int>& nums, int target) {// left + right = sum;// left - right = target;// left = (sum + target) / 2;// 01背包:背包总量left,在0-i个物品中,价值和为j的个数为dp[i][j]// dp[i][j] = dp[i-1][j]+dp[i-1][j-nums[i]]int len = nums.size();int sum = 0;for(int i = 0; i < len; i++) {sum = sum + nums[i];}if(sum < target || target < -sum) return 0;else if((sum - target) % 2 == 1) return 0;int left = (sum + target) / 2;vector<vector<int>> dp(len, vector<int>(left+1, 0));dp[0][0] = 1; // 初始化if(nums[0] <= left) dp[0][nums[0]]++; // 考虑left = nums[0] = 0的情况for(int i = 1; i < len; i++) {for(int j = 0; j <= left; j++) {if(j < nums[i]) dp[i][j] = dp[i-1][j];else dp[i][j] = dp[i-1][j] + dp[i-1][j-nums[i]];}}return dp[len-1][left];}
};