代码随想录算法训练营之JAVA|第二十一天| 77. 组合

今天是第21天刷leetcode,立个flag,打卡60天。

算法挑战链接

77. 组合icon-default.png?t=N6B9https://leetcode.cn/problems/combinations/description/

第一想法

需要从N个数中选取K个数,那么第一想法肯定是k个for循环,每个for循环选取一个数,比如从4个数中选取两个数,代码如下


for (int i = 1; i <= 4; i++) {

    for (int j = i + 1; j <= 4; j++) {
        //i 和 j 就是想要得到的两个数
    }
}

但是现在的问题是K是不确定的,也就是我需要在代码中写出K的for循环。 

直接写感觉是写不出来的,递归是一个好办法。

于是递归三要素就上阵了

  • 返回值和参数              返回一个列表, 参数: n, k, index,  列表
  • 结束递归的条件。      列表为null 或者 列表的size == k
  • 本层要完成的事情。   for循环index之后的数,并且添加到列表中,进入下一个递归

代码写出来之后会有一个很严重的问题,那就是列表会保留上一次结果添加的数。 

看完代码随想录之后的想法 

回溯也是递归的一种。但是回溯是比较特殊的一种,关于回溯的介绍在这里:回溯的介绍

 学习完成回溯之后,我们可以得出回溯的模版

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

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

于是我们按照模版来写出了下面的代码

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

    void backtracking(int n, int k,int index){
        if (result.size() == k) {
            results.add(new ArrayList<>(result));
            return;
        }

        for (int i = index; i <= n; i++){
            result.add(i);
            backtracking(n, k, i+1);
            result.removeLast();
        }

    }
}

实现过程中遇到哪些困难 

理解回溯是一件很困难的事情,主要困难的点在于回溯的思想是将解题过程看成是一颗树,穷举了所有的解体过程,然后先横向遍历,在纵向遍历,获取到每一种解题的可能。

今日收获

理解了回溯算法,并且拿到了回溯算法的模版

你可能感兴趣的:(算法,java,开发语言)