LeetCode 77. 组合

LeetCode 77. 组合
1、n和数中取k个数的组合：
图中每一个节点（图中为矩形），就代表本层的一个for循环。

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        backtracking(n,k,1);
        return result;
    }

private:
    //定义两个全局变量，分别用来存储结果集合和单个结果。
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(int n,int k, int startIndex){
        if(path.size() == k) {
            result.push_back(path);
            return;
        }
        for(int i = startIndex;i<=n;i++){
            path.push_back(i);// 处理当前节点
            backtracking(n,k,i+1);//递归处理下一个节点
            path.pop_back();// 回溯，撤销处理的节点
        }
    }
};
//走一走
//第一步将1插入到path中，然后遍历[2,3,4]
//第二步将2插入到path中，进入下一次递归，path满足条件，返回上层递归，然后将2pop出来
//第三步遍历第一个递归的[3,4]
...

2、优化：

如果n = 4，k = 4，那么图中打红x的部分是无效的

所以，可以剪枝的地方就在递归中每一层的for循环所选择的起始位置。
如果for循环选择的起始位置之后的元素个数 已经不足 我们需要的元素个数了，那么就没有必要搜索了。

接下来看一下优化过程如下：

已经选择的元素个数：path.size();

还需要的元素个数为: k - path.size();

在集合n中至多要从该起始位置 : n - (k - path.size()) + 1，开始遍历

为什么有个+1呢，因为包括起始位置，我们要是一个左闭的集合。

举个例子，n = 4，k = 3， 目前已经选取的元素为0（path.size为0），n - (k - 0) + 1 即 4 - ( 3 - 0) + 1 = 2。

从2开始搜索都是合理的，可以是组合[2, 3, 4]。

所以优化部分为

 for(int i = startIndex;i<=n - (k - path.size()) + 1;i++){

3、和为n的k个数的组合（组合中的数只包含0~9）

class Solution {
public:
    vector<vector<int>> combinationSum3(int k, int n) {
        backtracking(n,k,0,1);
        return result;
    }

private:
    vector<vector<int>> result;
    vector<int> path;
    void backtracking(int targetSum,int k,int sum,int startIndex){
        if(sum>targetSum) return;//剪枝操作
        if(path.size() == k){
            if(sum == targetSum) result.push_back(path);
            return;
        }
        for(int i = startIndex;i<=10-(k-path.size());i++){
            sum+=i;
            path.push_back(i);
            backtracking(targetSum,k,sum,i+1);
            sum-=i;
            path.pop_back();
        }
    }   
};

4、参考：
1、第77题. 组合
2、216.组合总和III