Leetcode 77 组合

题目

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

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

解题思路

  递归+回溯。注意剪枝的情况，即已经拿的数字个数 + 剩下数字个数 < k 的时候，后面的就不用判断了。

代码

class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> ans = new ArrayList<>();
        dfs(ans, new ArrayList<Integer>(), 1, n, k);
        return ans;
    }

    private void dfs(List<List<Integer>> ans, ArrayList<Integer> now, int pos, int n, int k) {
        if (now.size() == k) {
            ans.add(now);
            return;
        }
        for (int i = pos; i <= n; i++) {
            if (now.size() + n - i + 1 < k) break;
            now.add(i);
            dfs(ans, new ArrayList<Integer>(now), i + 1, n, k);
            now.remove(now.size() - 1);
        }
    }
}