leetcode刷题笔记【6】
类型:动态规划|记忆化搜索
494. 目标和【中等】
给你一个非负整数数组 nums 和一个整数 target 。
向数组中的每个整数前添加 ‘+’ 或 ‘-’ ,然后串联起所有整数,可以构造一个 表达式 :
例如,nums = [2, 1] ,可以在 2 之前添加 ‘+’ ,在 1 之前添加 ‘-’ ,然后串联起来得到表达式 “+2-1” 。
返回可以通过上述方法构造的、运算结果等于 target 的不同 表达式 的数目。
示例 1:
输入:nums = [1,1,1,1,1], target = 3
输出:5
解释:一共有 5 种方法让最终目标和为 3 。
-1 + 1 + 1 + 1 + 1 = 3
+1 - 1 + 1 + 1 + 1 = 3
+1 + 1 - 1 + 1 + 1 = 3
+1 + 1 + 1 - 1 + 1 = 3
+1 + 1 + 1 + 1 - 1 = 3
示例 2:
输入:nums = [1], target = 1
输出:1
提示:
1 <= nums.length <= 20
0 <= nums[i] <= 1000
0 <= sum(nums[i]) <= 1000
-1000 <= target <= 1000
//动态规划
//时间复杂度o(nm)
//空间复杂度o(nm)
//总和sum,正数a,负数的绝对值和就是sum-a
//a-(sum-a)= target ==》 a=(target+sum)/2
//即把题目变为从这些nums中挑选出a的方案数(01背包问题)
// class Solution {
// public int findTargetSumWays(int[] nums, int target) {
// int sum=0;
// for(int num:nums){
// sum+=num;
// }
// if((target+sum)<0||(target+sum)%2==1) return 0;
// int weight = (target+sum)/2;
// int[][] dp = new int[nums.length+1][weight+1];
// dp[0][0]=1;
// for(int i=1;i<=nums.length;i++){
// for(int j=0;j<=weight;j++){
// if(j>=nums[i-1])
// dp[i][j]=dp[i-1][j]+dp[i-1][j-nums[i-1]];
// else
// dp[i][j]=dp[i-1][j];
// }
// }
// return dp[nums.length][weight];
// }
// }
//记忆化搜索
class Solution {
int[][] memo;
public int findTargetSumWays(int[] nums, int target) {
int sum=0;
for(int num:nums){
sum+=num;
}
if((target+sum)<0||(target+sum)%2==1) return 0;
memo = new int[nums.length][(target+sum)/2+1];
for(int i=0;i<nums.length;i++)
Arrays.fill(memo[i],-1);
return f(nums,nums.length-1,(target+sum)/2);
}
int f(int[] nums,int i,int height){
if(i<0){
return height==0?1:0;
}
if(memo[i][height]!=-1) return memo[i][height];
if(height<nums[i]){
return memo[i][height]=f(nums,i-1,height); //不选
}
return memo[i][height]=f(nums,i-1,height)+f(nums,i-1,height-nums[i]);
}
}