第77题. 组合
给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
终于刷到回溯算法了。刚拿到题目还是蛮迷糊的,去网上看了一些例子再写,就比较舒服了。
package com.programmercarl.backtracking;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;
/**
* @ClassName Combine
* @Descriotion TODO
* @Author nitaotao
* @Date 2022/7/8 11:09
* @Version 1.0
**/
public class Combine {
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]
for (int i = t; i < n; i++) {
//如果元素已经添加,则跳过
if (list.contains(i)) {
continue;
}
//构建子树
list.add(i);
t++;
backtracking(n, k, t, list);
//删除最后一个结点,就是叶子结点
list.remove(list.size() - 1);
}
}
}
在多叉树遍历中,进行回溯。
如本题:如果回溯完毕,即 集合大小和列限制长度相等,即为一次结果集。
否则结果集未构成完毕,则需要进行回溯,通过循环,添加一个没有添加过的元素,构成回溯,然后再删除最后一个元素,退回来。继续添加元素进行回溯。
public void backtracking(参数) {
if (终止条件) {
存放结果;
return;
}
for (选择 : 本层集合中的元素(树中结点孩子的数量就是集合的大小)) {
处理结点;
//递归
backtracking(路径, 选择列表);
//回溯,撤销处理结果
}
}