【leetcode】Combinations (middle)

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],

]

 

思路：有点像0-1背包问题， 对于从1-n的每一个数字都可以选择放入答案 和不放入答案。 当长度达到k时就是一个符合条件的解。

递归的代码，AC了。只要注意状态的还原就好。

#include <iostream>

#include <vector>

#include <algorithm>

#include <queue>

#include <stack>

using namespace std;



class Solution {

public:

    vector<vector<int> > combine(int n, int k) {

        vector<vector<int>> ans;

        combinepart(ans, 1, k, n);

        return ans;

    }



    void combinepart(vector<vector<int>> &ans, int num, int k, int n)

    {

        static int i = 0;

        static vector<int> partans;

        if(num - 1 > n || partans.size() + (n - num + 2) < k) return; //数字超过了n 或者即使后面数字全部压入长度也不够的时候 直接返回 避免不必要的计算

        if(i == k)

        {

            ans.push_back(partans);

            return;

        }



        partans.push_back(num); //放入num

        i++;

        combinepart(ans, num + 1, k, n);



        partans.pop_back();

        i--;

        combinepart(ans, num + 1, k, n);//不放入num

    }

};



int main()

{

    Solution s;

    vector<vector<int>> ans = s.combine(3,3);



    return 0;

}

 

网上有非递归的代码，可是我好困，懒得看... 速度都差不多的，因为我的递归截枝了，没有多余的操作。

class Solution {

public:

    vector<vector<int> > combine(int n, int k) {

    vector<vector<int> >ret;

    if(k>n||k<=0)return ret;



    for(int i=1;i<=n-k+1;i++){

        ret.push_back(vector<int>(1,i));

    }



    for(int i=2;i<=k;i++){

        int num=ret.size();

        for(int j=0;j<num;j++){

            int last=ret[j].back();

            vector<int> pretmp=ret[j];

            ret[j].push_back(last+1);

            for(int p=last+2;p+k-i<=n;p++){

                vector<int> tmp=pretmp;

                tmp.push_back(p);

                ret.push_back(tmp);

            }

        }

    }



    return ret;

}

};