LeetCode 77. 组合

系列文章目录

回溯算法组合相关习题：LeetCode 77. 组合

---

---

前言

• 回溯算法之组合问题：N个数里面按一定规则找出k个数的集合

---

一、题目简介

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

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

示例一：

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

示例二：

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

二、思想逻辑

• 1、递归函数参数和返回值

result：存放符合条件结果的集合
pash：存放符合条件结果
stackIndex：记录下一层递归，搜索的起始位置
（一个集合来求组合的话，就需要startIndex）
（如果是多个集合取组合，各个集合之间相互不影响，那么就不用startIndex）

• 2、确定终止条件

确定参数：像[1,2]、[2，3]这种组合可以理解成一维数组，所以需要一个集合LinkedList<Integer>存放这些组合。
将组合的每一个个体放入一个新的结果集，也就是二维数组List<List<Integer>>存放这些组合的结果集。

• 3、确定单层递归逻辑

单层逻辑：stackIndex控制遍历的层数，当向pash集合中加入元素数量等同于k时，就将组合元素加入到result集合中。

完整代码

• 实现一：

 class Solution {
    List<List<Integer>> result = new ArrayList<>();
    LinkedList<Integer> pash = new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n, k, 1);
        return result;

    }
    public void backtracking(int n, int k, int stackIndex){
        if(pash.size() == k){
            result.add(new ArrayList<>(pash));
            return ;
        }
        for(int i = stackIndex; i <= n; i++){
            pash.add(i);
            backtracking(n, k, i + 1);
            pash.removeLast();
        }
    }
}

• 实现二：剪枝

class Solution {
    List<List<Integer>> result = new ArrayList<>();
    LinkedList<Integer> pash = new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        backtracking(n, k, 1);
        return result;
    }
    public void backtracking(int n, int k, int stackIndex){
        if(pash.size() == k){
            result.add(new ArrayList<>(pash));
            return;
        }
        //剪枝
        for(int i = stackIndex; i <= n - (k - pash.size()) + 1; i++){
            pash.add(i);
            backtracking(n, k, i + 1);
            pash.removeLast();
        }
    }
}

---