关闭

[leetcode] 77. Combinations

标签: leetcodebacktracing
218人阅读 评论(0) 收藏 举报
分类:

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

0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:68426次
    • 积分:3135
    • 等级:
    • 排名:第11664名
    • 原创:255篇
    • 转载:6篇
    • 译文:0篇
    • 评论:15条
    文章分类
    最新评论