目标和
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/target-sum
题目描述
给你一个整数数组 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
暴力递归
在递归时,因为每个数字都要选择,每个数字都有两种情况,一个是选择减,一个选择加,所以每个位置上都要两个可能,去不断递归。
代码演示
public int findTargetSumWays(int[] nums, int target) {
return process(nums,target,0);
}
/**
* 递归
*/
public int process(int[]nums,int target,int index){
//base case 数字选择完了,刚好组成目标数字,
//代表一个正确答案,返回1,否则返回0
if(index == nums.length){
return target == 0 ? 1 : 0;
}
//选择减号的情况
int p1 = process(nums,target - nums[index],index + 1);
//选择加号的情况
int p2 = process(nums,target + nums[index],index + 1);
return p1 + p2;
}
递归加缓存
/**
* 记忆化递归
*/
public int findTargetSumWays(int[] nums, int target) {
//缓存
HashMap<Integer,HashMap<Integer,Integer>>dp = new HashMap<>();
return process(nums,target,0,dp);
}
/**
* 递归加缓存
*/
public int process(int[]nums,int target,int index,HashMap<Integer,HashMap<Integer,Integer>>dp){
//缓存表中有 直接从缓存表中拿
if(dp.containsKey(index) && dp.get(index).containsKey(target)){
return dp.get(index).get(target);
}
int ans = 0;
if(index == nums.length){
ans = target == 0 ? 1 : 0;
}else{
int p1 = process(nums,target - nums[index],index + 1,dp);
int p2 = process(nums,target + nums[index],index + 1,dp);
ans = p1 + p2;
}
//index 位置不存在的话,就加进去
if(!dp.containsKey(index)){
dp.put(index,new HashMap());
}
dp.get(index).put(target,ans);
return ans;
}