LeetCode每日一题 2020-9-8 77.组合——回溯法dfs+剪枝

LeetCode每日一题 2020-9-8 77.组合——回溯法dfs+剪枝

题目描述
给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。

示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

解题思路
本题可用回溯法解决，可采用子集树的结构，如下图所示。可使用一棵高度为N+1子集数来表示从N个数中选择的所有情况，其中0代表不取这个数，1代表取这个数，最后的叶子结点便是选出来的结果。在回溯过程中，可以用一个vector记录已加入的数，当加入的数够题目中要求的k个时便可以结束。另外，使用回溯法要注意剪枝，否则会耗费过多的时间。本题中的剪枝函数可以是已加入的数的个数+剩余的树的层数 （意思就是现在已加入的+全部未加入的）小于k，便可以直接结束。例如，从10个数中选九个数，如果前两个数都不加入，剩下的就不用在判断了，因为就算剩下的8个数全部加入也小于9，这样能节约很多时间。

回溯法中使用子集树解题的代码框架：
void BackTrack(int t){
if(t>n) return; //结束条件
x[i]=1;
//对（t+1)的约束函数和限界函数
BackTrack(t+1);
x[i]=0;//还原
BackTrack(t+1);
}
题解代码

class Solution {
public:
    void dfs(int t,int n,int k,vector<int>& temp,vector<vector<int>> &ans){
        if(temp.size()>=k){
            ans.push_back(temp);
            return;
        }
        if(temp.size()+(n-t)<k-1){//剪枝，很重要，能省不少时间
            return;
        }
        if(t>n){
            return;
        }
        temp.push_back(t);
        dfs(t+1,n,k,temp,ans);
        temp.pop_back();
        dfs(t+1,n,k,temp,ans);
    }
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> ans;
        vector<int> temp;
        dfs(1,n,k,temp,ans);
        return ans;
    }
};