448,组合的几种解决方式_编程开发

We are all in the gutter, but some of us are looking at the stars.

身在井隅,心向璀璨。

问题描述

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

示例:


输入: n = 4, k = 2

输出:

[

[2,4],

[3,4],

[2,3],

[1,2],

[1,3],

[1,4],

]



回溯方式解决

这题要求的是从n个数字中选出k个弄成一组合,问最后总共有多少种组合,其实我们可以把它看做是一棵n-k+1叉树,比如n=4,k=2,那么就是一颗3叉树,这里以示例为例画个图看一下。


448,组合的几种解决方式_组合_02

注意这里只能选择后面的数字不能选择前面的数字,比如我选择了2,就能在选择1了,否则就会出现[1,2]和[2,1]这种重复的组合,同理我选择了3,那么1和2都不能再选了因为如果再选就会出现重复。


n叉树的遍历还记得吗

1private void backtrack() { 2    if ("终止条件") { 3        return; 4    } 5 6    for (int i = ?; i <= n - k + 1; i++) { 7        //逻辑运算1,(可有可无,视情况而定) 8 9        //递归调用,遍历每一个分支10        backtrack(list, n, k - 1, i + 1, tempList);1112        //逻辑运算2,(可有可无,视情况而定)13    }14}

我们来看下上面的框架,逻辑运算1的时候,就表示沿着当前分支走下去,我们把当前选择的值添加到集合中。逻辑2表示这个分支走完了我们要跳到另一个分支,之前讲​​426,什么是递归,通过这篇文章,让你彻底搞懂递归​​的时候提到过,从一个分支跳到另一个分支的时候,要么之前分支在运算之前先复制一份,要么把当前分支添加的值给移除,否则会造成分支污染。这两种方式都可以,但后一种效率会更好,他不会大量的复制数据。那么终止条件是什么呢,就是集合中的数据大小为k的时候,就表示找到了一组组合。搞懂了这一点代码就比较容易写了

1public List<List<Integer>> combine(int n, int k) { 2    List<List<Integer>> list = new LinkedList<>(); 3    backtrack(list, n, k, 1, new ArrayList<>()); 4    return list; 5} 6 7private void backtrack(List<List<Integer>> list, int n, int k, int start, List<Integer> tempList) { 8    //终止条件,找到一组组合 9    if (k == 0) {10        list.add(new LinkedList<>(tempList));11        return;12    }13    //注意这里的i不能从0开始,如果从0开始会出现重复的,比如[1,2]和[2,1]14    for (int i = start; i <= n - k + 1; i++) {15        //把当前值添加到集合中16        tempList.add(i);17        //递归调用18        backtrack(list, n, k - 1, i + 1, tempList);19        //从当前分支跳到下一个分支的时候要把之前添加的值给移除20        tempList.remove(tempList.size() - 1);21    }22}


参考二进制位

我们知道二进制位中,每个位置有两种状态,一种是0一种是1,这里也可以参照位运算的表示方式,每个数字都有选和不选两种状态,具体画个图来看下

448,组合的几种解决方式_编程开发_03

1public List<List<Integer>> combine(int n, int k) { 2    List<List<Integer>> list = new ArrayList<>(); 3    backtrack(list, n, k, 1, new ArrayList<>()); 4    return list; 5} 6 7private void backtrack(List<List<Integer>> list, int n, int k, int start, List<Integer> tempList) { 8    //终止条件,找到一对组合 9    if (k == 0) {10        list.add(new ArrayList<>(tempList));11        return;12    }13    if (start <= n - k) {14        //不选当前值,k不变15        backtrack(list, n, k, start + 1, tempList);16    }17    //选择当前值,k要减118    tempList.add(start);19    backtrack(list, n, k - 1, start + 1, tempList);20    //因为是递归调用,跳到下一个分支的时候,要把这个分支选的值给移除21    tempList.remove(tempList.size() - 1);22}


递归方式解决

这题要求的是从n个数字中选出k个弄成一组合,问最后总共有多少种组合,如果用数学知识就是C(n,k),这个公式又可以表示为C(n,k)=C(n-1,k-1)+C(n-1,k),也就是说要么选第n个数字,要么不选第n个数字。


(1),选第n个数字

如果选第n个数字,我们需要从前面n-1个数字中选择k-1个,然后在和数字n组合


(2),不选第n个数字

如果不选第n个数字,我们可以直接从前面n-1个数字中选择k个即可


那么最终的结果就是上面两种的和,我们来直接看下代码

1public List<List<Integer>> combine(int n, int k) { 2    List<List<Integer>> res = new ArrayList<>(); 3    //边界条件判断 4    if (n < k || k == 0) 5        return res; 6    //不选择第n个,从前面n-1个数字中选择k-1个构成一个集合 7    res = combine(n - 1, k - 1); 8    //如果res是空,添加一个空的集合 9    if (res.isEmpty())10        res.add(new ArrayList<>());11    //然后在前面选择的集合res中的每个子集合的后面添加一个数字n12    for (List<Integer> list : res)13        list.add(n);14    //res中不光要包含选择第n个数字的集合,也要包含不选择第n15    //个数字的集合16    res.addAll(combine(n - 1, k));17    return res;18}


总结

这题也不是特别难,但解题思路很多,每一种都比较经典。关于组合的题型也比较多,前面还讲过391,回溯算法求组合问题,有兴趣的也可以看下




448,组合的几种解决方式_组合_04

长按上图,识别图中二维码之后即可关注。


如果觉得有用就点个"赞"吧