Leetcode:Combinations

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

要产生全部的combinations,我们需要一种有序的方式产生combination以达到既产生全部又没有重复的目的。一个很简单的想法便是将[1,n]分成两个区间[1,k][k+1,n],然后交换两个区间中的1、2....min(k,n-k)个数。但这个方法,代码繁琐。在查找了网上的解答后,发现一种方法很好,代码简单,也易理解。

其主要想法是,先在[1,n]上抽象出一个有向图结构,如果节点i < j,那么有一条从i指向j的边。然后在这个有向图上,利用DFS,找出所有长度为k的path。

 1 class Solution {
 2 public:
 3     vector<vector<int> > combine(int n, int k) {
 4         vector<vector<int>> res;
 5         vector<int> path;
 6         dfs_combine(n,k,1,0,path,res);
 7         return res;
 8     }
 9     void dfs_combine(int n, int k, int start, int length, vector<int> & path, vector<vector<int>> & res){
10         if(length == k) res.push_back(path);
11         for(int i = start; i <= n; i++){
12             path.push_back(i);
13             dfs_combine(n,k,i+1,length+1,path,res);
14             path.pop_back();
15         }
16     }
17 };

 另一种递归的方法类似subset的生成,其实combination生成的是大小为k的subset,生成的思想是选当前元素和不选当前元素。代码如下:

class Solution {
public:
    vector<vector<int> > combine(int n, int k) {
        vector<vector<int> > result;
        vector<int> path;
        if(k > n) return result;
        
        get_combination(result, path, n, k, 1);
        
        return result;
    }
    
    void get_combination(vector<vector<int> > &result, vector<int> &path, int n, int k, int start){
        if(path.size() == k){
            result.push_back(path);
            return;
        }
        if(start > n) return;
        get_combination(result, path, n, k, start+1);//don't select start
        
        path.push_back(start);//select start
        get_combination(result, path, n, k, start+1);
        path.pop_back();
    }
};

迭代法:一个很简单的想法是从C(n, k-1)中生成C(n, k)。如何生成呢?很简单地,对于C(n, k-1)中的一个结果c, 我们从[1,n]中选出一个没有在c中出现的数字插入到c的末尾。但这里会产生重复的结果,为了避免重复,我们只选大于c中最大值并且[1,n]中没有出现的数字插入到c的末尾。代码如下:

class Solution {
public:
    vector<vector<int> > combine(int n, int k) {
        vector<vector<int> > result(1, vector<int>());
        
        for(int i = 1; i <= k; i++){
            vector<vector<int> > current;
            for(auto c:result){
                for(int j = (c.empty())?1:(*prev(c.end()) + 1); j <= n; j++){
                    if(find(c.begin(), c.end(), j) == c.end()){
                        current.push_back(c);
                        current.back().push_back(j);
                    }    
                }
            }
            result = current;
        }
        
        return result;
    }
};

 

posted on 2014-08-05 13:20  Ryan-Xing  阅读(119)  评论(0)    收藏  举报