代码随想录算法训练营第二十四天|77.组合

77.组合

题目描述

题目链接：力扣77.组合

给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。

你可以按 任何顺序 返回答案。

思路

对[1, n]的每个数都有两种可能，选 or 不选，可当作二叉树来思考：

根节点1选不选，选则进入左子树，不选则进入右子树。再依次判断每个数的可能性，同时记录已选的个数，达到k个就停止构建，记录当前组合，最终二叉树的每个叶子节点的路径都应该是一种组合。

回溯的思想。

代码实现

class Solution {
public:
    vector<int> temp;
    vector<vector<int>> ans;

    void dfs(int cur, int n, int k) {
        // 剪枝：temp 长度加上区间 [cur, n] 的长度小于 k，不可能构造出长度为 k 的 temp
        if (temp.size() + (n - cur + 1) < k) {
            return;
        }
        // 记录合法的答案
        if (temp.size() == k) {
            ans.push_back(temp);
            return;
        }
        // 考虑选择当前位置
        temp.push_back(cur);
        dfs(cur + 1, n, k);
        temp.pop_back();
        // 考虑不选择当前位置
        dfs(cur + 1, n, k);
    }

    vector<vector<int>> combine(int n, int k) {
        dfs(1, n, k);
        return ans;
    }
};