Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.
Example:
Input: n = 4, k = 2
Output:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
对于组合问题来说,顺序是不重要的。也就是说[1,2]和[2,1]是同一个元素,只关心选出来的是谁,而不关心这k个元素之间的顺序。
以n=4,k=2为例,按照题意,就是在[1,2,3,4]中取出两个数,在这种情况下,就是要先取出一个数,再取出一个数,
第1步取第一个数的时候,一共有4种可能,可以取1、可以取2、可以取3、也可以取4
假设取1的话,剩下的问题就变成在[2,3,4]中取1个数。此时就可以看到,存在一个递归的关系。两个红色的方框表达的是同一个问题,都是在一个集合中取一些数,只不过这个集合是谁不同,找的数不同而已。
在234中取一个数的话,比如说取2,那就得到了一个组合12,取3,就得到了13这个组合,取4就得到了14这个组合
如果在[1,2,3,4]中第一次取2的话,剩下的任务就是在[3,4]中取1个数。由于在组合中,不考虑取出元素的顺序,所以在取出元素时不再考虑1了。这是因为在前一个分支中,已经考虑了所得的组合中有1的情况。所以,取2个数,第一步取2的话,剩下一个数只考虑在3、4两个数中取就好了。因为再取1的话,得到的是21,和前一个分支中得到的12是一样的。
所以在这个分支中,在3、4中取值,如果取3,得到的是23,如果取4,得到的是24
在[1,2,3,4]中取2个数,第1步如果取3的话,那么剩下的问题就是只能在[4]这一个数字中取一个数字,这里依然是取1、取2这种可能,在前面两个分支中已经考虑过了。所以取3以后,只需要在3后面的数字中考虑就好了。3后面的数字只有4一个,那么就只考虑它就好了。
此时只能取4,得到34这样一个组合
最后,如果一上来就取4的话,那么剩下的数字就没有了,所以在剩下的数字中再取一个数是不可能的。所以,在这个分支里是产生不了解的。此时就得到了在[1,2,3,4]中取两个数的所有的组合。这个数虽然看起来和之前的数不太一样,因为之前的数每个节点有几个孩子是固定的,但是这棵树有几个孩子不固定,不过,它依然是这样的一个树形的结构。依然可以使用递归来解决这个问题。在解决问题的过程中,依然可以看到回溯这样的过程。换句话说,第1步取1,第二步依次取2、3、4,得到了存在1的所有的组合,之后,就要回到在[1,2,3,4]中取2个数的这个问题中,再继续尝试取2这个数。当第1步取2的时候,第二步只能在[3,4]中取1个数,当尝试了取3、取4之后,又要回到在[1,2,3,4]中取2个数这样一个方法,尝试取3,依次类推。那么,这样就是一个回溯的问题。
代码:
class Solution {
private List<List<Integer>> res=new ArrayList();
public List<List<Integer>> combine(int n, int k) {
if(n<=0||k<=0||k>n)
return res;
getCombine(n,k,1,new ArrayList());
return res;
}
private void getCombine(int n,int k,int start,List<Integer> curlist){
if(curlist.size()==k){
res.add(curlist);
return;
}
for(int i =start;i<=n;i++){
ArrayList<Integer> temp=new ArrayList();
temp.addAll(curlist);
temp.add(i);
getCombine(n,k,i+1,temp);
}
}
}
剪枝后的解法:
class Solution {
private List<List<Integer>> res=new ArrayList();
public List<List<Integer>> combine(int n, int k) {
if(n<=0||k<=0||k>n)
return res;
getCombine(n,k,1,new ArrayList());
return res;
}
private void getCombine(int n,int k,int start,List<Integer> curlist){
if(curlist.size()==k){
res.add(curlist);
return;
}
for(int i =start;i<=n-(k-curlist.size())+1;i++){
ArrayList<Integer> temp=new ArrayList();
temp.addAll(curlist);
temp.add(i);
getCombine(n,k,i+1,temp);
}
}
}