Leetcode 39. 组合总数(1 2) tag 数组 回溯法

一、给定一个无重复元素的数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的数字可以无限制重复被选取。

说明:

  • 所有数字(包括 target)都是正整数。
  • 解集不能包含重复的组合。 

示例 1:

输入: candidates = [2,3,6,7], target = 7,
所求解集为:
[
  [7],
  [2,2,3]
]

示例 2:

输入: candidates = [2,3,5], target = 8,
所求解集为:
[
  [2,2,2,2],
  [2,3,3],
  [3,5]
]
思路:使用深度遍历,遍历时可以不选择该数直接选择k+1,也可以选择k数,并且可以重复选择
class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        vector<int> tmp;
        dfs(candidates,target,0,0,tmp);
        return res;
    }
    void dfs(vector<int>& candidates, int target, int k, int sum, vector<int>& tmp)
    {
        if (sum>=target)
        {
            if(sum==target)
                res.push_back(tmp);
            return;
        }
        if (k>=candidates.size())
            return;
        // 选择k,因为可以重复选择,所以k不加1
        if(sum+candidates[k]<=target)
        {
            tmp.push_back(candidates[k]);
            dfs(candidates,target,k,sum+candidates[k],tmp);
            tmp.pop_back();
        }
        // 不选择k,直接进入下一个数
        dfs(candidates,target,k+1,sum,tmp);
    }
    vector<vector<int>> res;
};

 二、

给定一个数组 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。candidates 中的每个数字在每个组合中只能使用一次。

说明:

所有数字(包括目标数)都是正整数。
解集不能包含重复的组合。

示例 1:

输入: candidates = [10,1,2,7,6,1,5], target = 8,
所求解集为:
[
[1, 7],
[1, 2, 5],
[2, 6],
[1, 1, 6]
]
示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
所求解集为:
[
[1,2,2],
[5]
]

class Solution {
public:
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        int n=candidates.size();
        if(n<1)
            return res;
        sort(candidates.begin(),candidates.end());
        vector<int> tmp;
        dfs(candidates,target,0,0,tmp);
        return res;
    }
    void dfs(vector<int>& candidates, int target, int k, int sum, vector<int>&tmp)
    {
        if (sum >=target)
        {
            if(sum==target)
                res.push_back(tmp);
            return;
        }
        if (k >= candidates.size())
            return;
        for(int i=k;i<candidates.size(); ++i)
        {
            // 由于不能有重复序列,因此有重复数字时不要重复放入
            if (i>k&&candidates[i]==candidates[i-1])
                continue;
            tmp.push_back(candidates[i]);
            dfs(candidates,target,i+1,sum+candidates[i],tmp);
            tmp.pop_back();
        }
    }
    vector<vector<int>> res;
};

 

posted @ 2021-03-03 21:00  鸭子船长  阅读(69)  评论(0编辑  收藏  举报