494. 目标和
思路:方法一,枚举,位运算。该方法超时了,但还是贴一下
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int n=nums.size();
int ans=0;
for(int i=0;i<1<<n;i++){
int t=0;
for(int j=0;j<n;j++){
if(i>>j&1) t+=nums[j];
else t-=nums[j];
}
if(t==target) ans++;
}
return ans;
}
};
思路:方法二,哈希表,滚动数组。因为数组nums的和不超过1000,那么对所有元素操作后的范围在-1000~1000之间。那么我们就可以利用哈希表a来记录对数组nums操作之后,各种结果出现的次数。
class Solution {
public:
int findTargetSumWays(vector<int>& nums, int target) {
int n=nums.size();
//用到滚动数组
int a[2][2020];
memset(a,0,sizeof a);
//为了处理负数,在所有可能值的基础上+1000
a[0][1000]=1;
int t=1;//实现滚动数组的效果
for(int i=0;i<n;i++){
t=1-t;
memset(a[1-t],0,sizeof a[1-t]);
for(int j=0;j<2020;j++){
if(a[t][j]>0){
if(j+nums[i]<2020) a[1-t][j+nums[i]]+=a[t][j];
if(j-nums[i]>=0) a[1-t][j-nums[i]]+=a[t][j];
}
}
}
return a[1-t][target+1000];
}
};