Combinations - LeetCode

[toc]

題目連接

Combinations - LeetCode

注意點

解法

解法一：遞歸，若是tmp中有k個數字就把tmp加到ret中，不然繼續往tmp添加數字。

class Solution {
public:
    void recursion(int n,int k,int start,vector<int> tmp,vector<vector<int>>& ret)
    {
        if(tmp.size() == k) 
        {
            ret.push_back(tmp);
            return;
        }
        int i;
        for(i = start;i <= n;i++)
        {
            tmp.push_back(i);
            recursion(n,k,i+1,tmp,ret);
            tmp.pop_back();
        }
    }
    vector<vector<int>> combine(int n, int k) {
        vector<vector<int>> ret;
        vector<int> tmp;
        recursion(n,k,1,tmp,ret);
        return ret;
    }
};

解法二：非遞歸，若是tmp中有k個數字就把tmp加到ret中，不然繼續往tmp添加數字。

class Solution {
public:
    vector<vector<int> > combine(int n, int k) {
        vector<vector<int> >ret;
        vector<vector<int> >vec(1);
        for(int i = 1; i <= n; i++)
        {
            int len = vec.size();
            vector<int> tmp;
            for(int j = 0; j < len; j++)
            {
                tmp = vec[j];
                tmp.push_back(i);
                if(tmp.size() == k)ret.push_back(tmp);
                else vec.push_back(tmp);
            }
        }
        return ret;
    }
};

小結

• 怎麼都這麼耗時呢