day24 回溯 | 77

77. 组合

经典的组合问题，使用递归 + 回溯的方式进行解决

package algor.trainingcamp;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * @author lizhe
 * @version 1.0
 * @description: 组合问题
 * @date 2023/4/28 09:33
 *
 * 给定两个整数 n 和 k，返回范围[1, n] 中所有可能的 k 个数的组合。
 *
 * 你可以按 任何顺序 返回答案。
 *
 * 回溯 + 递归
 *
 * 优化点： 如果for循环选择的起始位置之后的元素个数 已经不足 我们需要的元素个数了，那么就没有必要搜索了。
 */
public class LeetCode77 {
    LinkedList<Integer> subList = new LinkedList<>();
    List<List<Integer>> res = new ArrayList<>();

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

    private void helper(int n, int k, int startIdx) {
        //递归截止条件
        if(subList.size() == k){
            res.add(new ArrayList<>(subList));
            return;
        }

        /**
         * 剪枝:
         * * 已经选择的元素个数: subList.size()
         * * 待选择的是: k - subList.size()
         * * 开始位置: n - (k - subList.size()) + 1
         */
        for(int i = startIdx;i <= n - (k - subList.size()) + 1;i++){
            subList.add(i);
            helper(n,k,i + 1);
            subList.removeLast();
        }
    }
}

回溯模板:

void  backtracking(){

	if(终止条件)	{

		收集结果 

		return

	}

for(集合的元素集，类似子节点的个数)

	{

		处理结点

		递归函数；

		回溯操作

	（撤销处理结点12， 2撤销 ，13 撤销3， 14）

	}

}