代码随想录算法训练营day24|回溯算法

最新推荐文章于 2024-09-03 18:29:17 发布

Saiyan99

最新推荐文章于 2024-09-03 18:29:17 发布

阅读量54

点赞数

文章标签：算法 leetcode 职场和发展

本文链接：https://blog.csdn.net/Saiyan99/article/details/130042020

版权

文章介绍了如何使用回溯算法解决LeetCode中的组合问题，初始代码未进行剪枝操作，导致效率较低。通过添加剪枝条件，优化搜索起点，提高算法性能。在回溯过程中，当路径长度等于目标长度时添加到结果列表，然后回溯并移除路径中的最后一个元素，避免重复。

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

没接触过回溯算法，第一次遇见有点模糊的感觉，需要多加练习

2023.4.7 组合

77. 组合 - 力扣（LeetCode）

没有进行剪枝操作的代码。

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    LinkedList<Integer> path = new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        backTracking(n,k,1);
        return res;
    }
    public void backTracking(int n,int k,int start){
        if(path.size() == k){
            res.add(new ArrayList(path));
            return;
        }
        for(int i = start;i <= n;i++){
            //path路径还没有满足要求，直接放入path中
            path.add(i);
            //进入递归,这里start需要+1，是因为我们放入的值时不能重复的；
            backTracking(n,k,i+1);
            //退出递归，代表我们的path长度已经满足了要求，此时需要减去最后一个元素，重新放入新的元素
            path.removeLast();
        }
    }
}

我们其实可以发现如果start到末尾的位置元素个数少于了k，其实就没有意义了，不需要再对其进行统计，因此可以对这部分内容进行优化。

例子：假设n=7，k=4，那么从5开始搜索就没有意义了，因此我们的start的搜索起点是存在一个小于n的界限的，

即搜索起点的上界+接下来要选择的元素个数-1 = n

而我们接下来要选择的元素个数 = k-path.size()

因此我们搜索起点的上界 = n-(k-path.size())+1

因此我们的for循环判断关系就从i < n变为了 i < n - ( k - path.size()) + 1

class Solution {
    List<List<Integer>> res = new ArrayList<>();
    LinkedList<Integer> path = new LinkedList<>();
    public List<List<Integer>> combine(int n, int k) {
        backTracking(n,k,1);
        return res;
    }
    public void backTracking(int n,int k,int start){
        if(path.size() == k){
            res.add(new ArrayList(path));
            return;
        }
        for(int i = start;i <=  n - ( k - path.size()) + 1;i++){
            //path路径还没有满足要求，直接放入path中
            path.add(i);
            //进入递归,这里start需要+1，是因为我们放入的值时不能重复的；
            backTracking(n,k,i+1);
            //退出递归，代表我们的path长度已经满足了要求，此时需要减去最后一个元素，重新放入新的元素
            path.removeLast();
        }
    }
}