494. Target Sum 494.目标总和

You are given a list of non-negative integers, a1, a2, ..., an, and a target, S. Now you have 2 symbols + and -. For each integer, you should choose one from + and - as its new symbol.

Find out how many ways to assign symbols to make sum of integers equal to target S.

Example 1:

Input: nums is [1, 1, 1, 1, 1], S is 3. 
Output: 5
Explanation: 

-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

There are 5 ways to assign symbols to make the sum of nums be target 3.

思路:试试回溯:
dfs(int[] nums, int S, )
数量不知道该放在哪里
其实为了凑出来一个和,参数里有个pos, 然后sum + nums[pos]就行了

 

return ; //position到头之后,无论如何都要退出一下

 

两个DFS都进行就行:

dfs(nums, pos + 1, currentSum + nums[pos], S);
dfs(nums, pos + 1, currentSum - nums[pos], S);

 

class Solution {
    int count;
    
    public int findTargetSumWays(int[] nums, int S) {
        //cc
        if (nums == null || nums.length == 0)
            count = 0;
        
        //dfs
        dfs(nums, 0, 0, S);
        
        return count;
    }
    
    public void dfs(int[] nums, int pos, int currentSum, int S) {
        //exit
        if (pos == nums.length) {
            if (currentSum == S) 
                count++;
            return ; //无论如何都要退出一下
        }
        
        dfs(nums, pos + 1, currentSum + nums[pos], S);
        dfs(nums, pos + 1, currentSum - nums[pos], S);
    }
}
View Code

 


 

posted @ 2020-07-25 11:16  苗妙苗  阅读(97)  评论(0编辑  收藏  举报