leetcode刷题：回溯算法02（组合优化）

第77题. 组合

力扣题目链接

给定两个整数 n 和 k，返回 1 … n 中所有可能的 k 个数的组合。

示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]

算法优化，进行了剪枝操作。
就是在循环时，
已经选择的元素个数为： list.size();
还需要的元素个数为：k - path.size()
在集合n中至多要从该起始位置：n - ( k - path.size())+1 开始遍历
+1是因为起始位置。这是左闭的集合。

package com.programmercarl.backtracking;
import java.util.ArrayList;
import java.util.List;

/**
 * @ClassName CombingAdv
 * @Descriotion TODO
 * @Author nitaotao
 * @Date 2022/7/8 14:54
 * @Version 1.0
 * https://leetcode.cn/problems/combinations/submissions/
 * 77. 组合 优化
 **/
public class CombingAdv {
    List<List<Integer>> result = new ArrayList<>();

    public static void main(String[] args) {
        new Combine().combine(4, 2);
    }

    public List<List<Integer>> combine(int n, int k) {
        List<Integer> list = new ArrayList<>();
        // [1,n]
        backtracking(n, k, 1, list);
        System.out.println(result);

        return result;
    }

    /**
     * @param n    数字范围
     * @param k    列表限制长度
     * @param t    为了防止全排列，给的范围
     * @param list 结果集
     *             剪枝 操作
     */
    public void backtracking(int n, int k, int t, List<Integer> list) {

        //如果找到结果集，就添加
        // k为结果集的长度
        if (list.size() == k) {
            result.add(new ArrayList<>(list));
            return;
        }
        // [1,n]
        // 此处不用循环 n - t 次
        // 因为只需要 k 个元素，所以 k 次即可。
        // i + k <= n
        // i <= n - k
        // 一共需要 n^2 *（n - k）！ 次
        // 每次已进行了 list.size() 次回溯
        // 还需要  k-list.size()-1 次
        // 所以只需要移动  n - ( k - list.size()) + 1 次
        // 1 为起始位置   i + k = n
        for (int i = t; i <= n - (k - list.size()) + 1; i++) {
            //如果元素已经添加，则跳过
            if (list.contains(i)) {
                continue;
            }
            //构建子树
            list.add(i);
            t++;
            backtracking(n, k, t, list);
            //删除最后一个结点，就是叶子结点
            list.remove(list.size() - 1);
        }
    }
}