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

阅读更多
版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/TstsUgeg/article/details/49909165
文章标签: leetcode backtracing
个人分类: leetcode
想对作者说点什么? 我来说一句

没有更多推荐了,返回首页

不良信息举报

[leetcode] 77. Combinations

最多只允许输入30个字

加入CSDN,享受更精准的内容推荐,与500万程序员共同成长!
关闭
关闭