[leetcode] 77. Combinations

Given two integers n and k, return all possible combinations of k numbers out of 1 ... n.

For example,
If n = 4 and k = 2, a solution is:

[
  [2,4],
  [3,4],
  [2,3],
  [1,2],
  [1,3],
  [1,4],
]

这道题其实就是列出C(n,k)的所有情况，难度为Medium。

比较直观的想法是采用递归回溯遍历每个数字取和不取的情况，递归结束条件是已经取到了k个数字，注意回溯之后需要修改的标记。具体代码：

class Solution {
public:
    vector<vector<int> > combine(int n, int k) {
        vector<vector<int>> rst;
        vector<int> curRst;
        
        if(k > 0 && n >= k) 
    ​    ​    ​getCombine(rst, curRst, 0, 0, n, k);
        
        return rst;
    }
    
    void getCombine(vector<vector<int>> &rst, vector<int> &curRst, int pos, int cnt, int n, int k) {
        if(cnt == k) rst.push_back(curRst);
        else {
            if(n-pos < k-cnt-1) return;
            else {
                while(pos < n) {
                    curRst.push_back(pos+1);
                    pos++;
                    cnt++;
                    getCombine(rst, curRst, pos, cnt, n, k);
                    cnt--;
                    curRst.pop_back();
                }
            }
        }
    }
};

另外还可以根据公式C(n,k) = C(n-1,k-1) + C(n-1,k)，用纯递归的方法解决此问题。这里采用的其实是分治法，根据取与不取第n个数分为C(n-1,k-1)和C(n-1,k)，然后将两部分合并。需要注意的是在得到C(n-1,k-1)之后要在每个条目中加入n，因为C(n-1,k-1)是默认选取n的，具体代码：

class Solution {
public:
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> rst1, rst2;
        
        if(n<k || k<0) return rst1;
        if(n==k || k==0) {
            vector<int> curRst;
            for(int i=1; i<=k; i++) curRst.push_back(i);
            rst1.push_back(curRst);
            return rst1;
        }
        
        rst1 = combine(n-1, k-1);
        for(int i=0; i<rst1.size(); i++) 
    ​    ​    ​rst1[i].push_back(n);
        
        rst2 = combine(n-1, k);
        
        copy(rst2.begin(), rst2.end(), back_inserter(rst1));
        
        return rst1;
    }
};