代码随想录算法训练营第24天｜77. 组合

Day 24

77. 组合

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

    void dfs(int cur, int n, int k) {

        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> combine(int n, int k) {
        dfs(1, n, k);
        return ans;
    }
};

java版：

class Solution {
    List temp = new ArrayList();
    List> ans = new ArrayList>();

    public List> combine(int n, int k) {
        dfs(1, n, k);
        return ans;
    }

    public 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.add(new ArrayList(temp));
            return;
        }
        // 考虑选择当前位置
        temp.add(cur);
        dfs(cur + 1, n, k);
        temp.remove(temp.size() - 1);
        // 考虑不选择当前位置
        dfs(cur + 1, n, k);
    }
}