77. 组合

最新推荐文章于 2024-08-29 16:23:51 发布

ceresyezi

最新推荐文章于 2024-08-29 16:23:51 发布

阅读量46

点赞数

分类专栏：刷题总结 # 回溯文章标签：深度优先 leetcode 算法

本文链接：https://blog.csdn.net/qq_41830037/article/details/130506773

版权

刷题总结同时被 2 个专栏收录

110 篇文章 0 订阅

订阅专栏

回溯

12 篇文章 0 订阅

订阅专栏

该文章介绍了一种使用Java实现的回溯算法来解决LeetCode上的组合问题。给定两个整数n和k，程序返回[1,n]中所有可能的k个数的组合。关键在于递归过程中通过for循环的剪枝策略，避免不必要的搜索。

摘要由CSDN通过智能技术生成

https://leetcode.cn/problems/combinations/

题目要求

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

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

回溯

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    LinkedList<Integer> list = new LinkedList<>();

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

    public void dfs(int index, int n, int k) {
        // 递归结束条件
        if (list.size() == k) {
            res.add(new ArrayList<>(list));
            return;
        }
        // 循环 回溯
        for (int i = index; i <= n - (k - list.size()) + 1; i++) {
            list.add(i);
            dfs(i + 1, n, k);
            list.removeLast();
        }
    }
}

总结

定义全局变量，可以减少函数中传递的参数，提高可读性；
剪枝的地方就在递归中每一层的for循环所选择的起始位置。
- 如果for循环选择的起始位置之后的元素个数已经不足我们需要的元素个数了，那么就没有必要搜索了。
- for (int i = index; i <= n - (k - list.size()) + 1; i++) // i为本次搜索的起始位置
- +1因为包括起始位置，我们要是一个左闭的集合。