给定一个非负整数数组,a1, a2, ..., an, 和一个目标数,S。现在你有两个符号 +
和 -
。对于数组中的任意一个整数,你都可以从 +
或 -
中选择一个符号添加在前面。
返回可以使最终数组和为目标数 S 的所有添加符号的方法数。
示例:
输入:nums: [1, 1, 1, 1, 1], S: 3
输出:5
解释:
-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
一共有5种方法让最终目标和为3。
提示:
- 数组非空,且长度不会超过 20 。
- 初始的数组的和不会超过 1000 。
- 保证返回的最终结果能被 32 位整数存下。
链接:https://leetcode-cn.com/problems/target-sum/
思路
思路有点类似跨阶梯的题目,需要记录到前一阶梯的行走情况,这一阶梯根据此变换。
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int S) {
int num = nums.size();
if(num==0) return 0;
unordered_map<int, int> rec;
rec[0] = 1;
for(int i=0; i<nums.size(); ++i){
unordered_map<int, int> tmp;
for(auto it: rec){
tmp[it.first+nums[i]] += it.second;
tmp[it.first-nums[i]] += it.second;
}
rec = tmp;
}
auto res = rec.find(S);
if(res!=rec.end()){
return res->second;
}
return 0;
}
};
map的空间复杂度有点高,可以根据数据边界,使用数组。