标签：
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; } };
【leetcode】Combinations (middle)
标签：
原文地址：http://www.cnblogs.com/dplearning/p/4237673.html
