代码随想录 第二十四天| ●回溯 理论基础 ● 77. 组合

回溯理论基础:

  回溯三部曲:制定回溯函数的参数和返回值

        确定回溯终止条件

        确定回溯遍历过程

 

回溯模板

void backtracking(参数) {
    if (终止条件) {
        存放结果;
        return;
    }

    for (选择:本层集合中元素(树中节点孩子的数量就是集合的大小)) {
        处理节点;
        backtracking(路径,选择列表); // 递归
        回溯,撤销处理结果
    }
}

leetcode:77. 组合 - 力扣(LeetCode)

class Solution {
    List<List<Integer>> result= new ArrayList<>();
    LinkedList<Integer> path = 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 startIndex){
        //等于k就是一个组合,直接存到res数组
        if (path.size() == k){
            result.add(new ArrayList<>(path));
            return;
        }
        //回溯
        for (int i =startIndex;i<=n;i++){
            path.add(i);
            backtracking(n,k,i+1);
            path.removeLast();
        }
    }
}

 剪枝

public void backtracking(int n,int k,int startIndex){
        //等于k就是一个组合,直接存到res数组
        if (path.size() == k){
            result.add(new ArrayList<>(path));
            return;
        }
        //回溯
        for (int i =startIndex;i<=n;i++){
            //剪枝操作,path存的加上剩余的数(n-i+1)就是目前的总数
            if(path.size() + (n - i + 1) < k){
                continue;
            }
            path.add(i);
            backtracking(n,k,i+1);
            path.removeLast();
        }
    }

 

posted @ 2024-03-20 13:24  22软工冷薄  阅读(1)  评论(0编辑  收藏  举报