LeetCode 77. 组合

77. 组合

难度 中等

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

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

示例 1：

输入：n = 4, k = 2
输出：
[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

示例 2：

输入：n = 1, k = 1
输出：[[1]]

提示：

• 1 <= n <= 20
• 1 <= k <= n

题解

这是道组合题目，一般采用方法有栈或者递归。比较常用递归，这里就用递归。那怎么递归调用呢，这里采用枚举去尝试，每次都加入新的数字，看看啥时候枚举到k个。

dfs函数有几个重要的参数

• int begin，这是枚举的起始下标（因为这是组合题目，不是排列题目，前面枚举过的数，后面不能再重复枚举）起始下标保证枚举顺序
• int n，从多少数里枚举
• int k，枚举多少个数，每轮枚举完，k都会减一；当减到零，枚举成功，存储结果
• Deque queue，队列，方便操作
• List<List> ans，存储结果

class Solution {
    public List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> ans = new ArrayList<List<Integer>>();//存储结果
        Deque<Integer> queue = new ArrayDeque<>(k);//队列，方便入队出队
         for(int i = 1; i <= n; i++){//i为枚举遍历，从1开始到n
             queue.addLast(i);//i入队
             dfs(i, n, k - 1, queue, ans);//递归调用枚举
             queue.removeLast();//i出队，还原
         }
         return ans;
    }
    void dfs(int begin, int n, int k, Deque<Integer> queue, List<List<Integer>> ans){
        if(k == 0){//k等于0，枚举成功
            ans.add(new ArrayList<Integer>(queue));//加入队列
            return;
        }
        if(begin > n - k + 1){//剪枝操作，提升速度，当起始坐标已经操作能枚举的下限，从当前begin开始枚举不出k个数，结束递归
            return ;
        }
        for(int i = begin + 1; i <= n; i++){//从begin+1开始枚举，即改变枚举起始坐标
            queue.addLast(i);//i入队
            dfs(i, n, k - 1, queue, ans);//递归调用枚举
            queue.removeLast();//i出队，还原
        }
    }
}