LeetCode 39 40 组合总和 组合总和II (回溯)

1.组合总和

难度:中等
给定一个无重复元素的数组 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]
]

思路:第一次做回溯的算法题,摸不清楚套路,因此看了网上的思路

class Solution {
public:
    vector<vector<int>> combinationSum(vector<int>& candidates, int target) {
        vector<int> items;
        vector<vector<int>> result;
        backtrade(candidates,0,target,result,items);
        return result;
    }
    
    void backtrade(vector<int>& candidates, int start, int target, vector<vector<int>> &result, vector<int> items){
        if(target<0) return; //剪支
        if(target==0) result.push_back(items);
        for(int i=start;i<candidates.size();i++){ //从start开始,避免重复
            items.push_back(candidates[i]);
            backtrade(candidates,i,target-candidates[i],result,items);
            items.pop_back();
        }
    }
};

2.组合总和II

难度:中等
给定一个数组 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) {
        vector<vector<int>> result;
        vector<int> items;
        sort(candidates.begin(),candidates.end());
        backtrade(candidates,0,target,result,items);
        return result;
    }
    
    void backtrade(vector<int>& candidates,int start,int target,vector<vector<int>>& result,vector<int> items){
        if(target<0) return;
        else if(target==0) result.push_back(items);
        else {
            for(int i=start;i<candidates.size();i++){
                if(i>start && candidates[i]==candidates[i-1]) continue;
                items.push_back(candidates[i]);
                backtrade(candidates,i+1,target-candidates[i],result,items);
                items.pop_back();
            }
        }
    }
    
};

猜你喜欢

转载自blog.csdn.net/Hansry/article/details/83539084
今日推荐