LeetCode 77. Combinations

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);
        }
        
    }
}