LeetCode算法入门之 递归 / 回溯——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. 标签：回溯与剪枝 
 2. n 表示范围为 1...n，k 表示剩余空间，start 表示开始位置，temp 为回溯列表 
 3. 判断 k== 0，如果为 0 则代表 temp 中已经存入 k 个数，拷贝 temp 存入结果 ans 中
 4.  如果不为 0，从 start 位置开始递归调用，
 	现将当前位置数据加入 temp 中，并进入下一层，
  	等待返回后将本层加入的数据移除，本质就是树的构造过程
 5. 其中循环结束条件默认为最大值到 n，
 	这里可以优化进行剪枝，比如 n=4，k=3 时，
 	如果列表从 start=3 也就是 [3] 开始，那么该组合一定不存在，
 	因为至少要 k=3 个数据，所以剪枝临界点为 n-k+1

代码

class Solution {

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

   public void dfs(int start,int n,int k){
   //剪枝
       if(k==0){
           ans.add(new ArrayList<>(temp));
           return; 
       }
       for(int i=start;i<=n-k+1;i++){
           temp.add(i);
           dfs(i+1,n,k-1);
           temp.remove(temp.size()-1);
       }
   }
}