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

108 阅读1分钟

从零开始学习c++,每天起码做一道leetcode题目,在此记录,希望最后能够有所收获!

一、题目描述

40.组合总和II

给定一个候选人编号的集合 candidates 和一个目标数 target ,找出 candidates 中所有可以使数字和为 target 的组合。

candidates 中的每个数字在每个组合中只能使用 一次 。

注意:解集不能包含重复的组合。 

示例 1:

输入: candidates = \[10,1,2,7,6,1,5], target = 8,
输出:
[
[1,1,6],
[1,2,5],
[1,7],
[2,6]
]

示例 2:

输入: candidates = [2,5,2,1,2], target = 5,
输出:
[
[1,2,2],
[5]
]

提示:

  • 1 <= candidates.length <= 100
  • 1 <= candidates[i] <= 50
  • 1 <= target <= 30

二、思路分析

这道题目的关键之处在于两点,第一点是candidates中的每个数字在每个组合中只能使用一次;第二点是解集中不能包含重复的组合。

首先来看第一点,每个数字在一个组合中都只能重复一次。回溯这种算法是逮着一条路一直往下走,直到遇见一个合适的结果后,记录下该结果,再进行回溯,返回上一个节点。为了保证每个数字在一个组合中只出现一次,就使用了start来,每次递归时传递的参数为i+1,。

第二点,解集不能包含重复的元素,但题目所给的例子中有[10,1,2,7,6,1,5]这样的,1重复的出现,假如target为6,那么按照之前的思路就会出现两个[1,5]。怎么避免?首先对数组进行排序,这样数组就变成了[1,1,2,5,6,7,10],用一个record数组来记录每个位置的数是否出现过。这样就能解决该问题。

三、AC代码

class Solution {
public:
vector\<vector<int>>ans;
vector<int>path;
vector<int>record=vector<int>(101,0);
void backtracking(vector<int> candidates,int target,int start){

        if(accumulate(path.begin(),path.end(),0)==target){
            ans.push_back(path);
            return;
        }

        for(int i=start;i<candidates.size();i++){
            if(i>0&&candidates[i]==candidates[i-1]&&record[i-1]==0){
                continue;
            }
            if((accumulate(path.begin(),path.end(),0)+candidates[i])>target){
                continue;
            }
            path.push_back(candidates[i]);
            record[i]=1;
            backtracking(candidates,target,i+1);
            record[i]=0;
            path.pop_back();

        }
    }
    vector<vector<int>> combinationSum2(vector<int>& candidates, int target) {
        sort(candidates.begin(),candidates.end());
        backtracking(candidates,target,0);
        return ans;

    }

};

四、总结

这种情况就需要进行去重了