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],
]
题目是求组合数。
用递归的思想。
下面是我的第一份代码。
class Solution {
public:
vector<vector<int>> ans;
vector<int> v;
vector<vector<int>> combine(int n, int k) {
if(n==0||k==0)
return ans;
dfs(1,0,n,k);
return ans;
}
void dfs(int t,int num,int n,int k){
if(num==k){
ans.push_back(v);
return;
}
for(int i=t;i<=n;i++){
v.push_back(i);
dfs(1+t,num+1,n,k);
v.pop_back();
}
}
};
运行结果让我很意外,居然超时了。
Submission Result: Time Limit Exceeded More Details
Last executed input:
10
7
为了搞清楚问题所在,我就去写了个测试代码。
#include <iostream>
#include <cstdio>
#include <vector>
using namespace std;
vector<vector<int> > ans;
vector<int> v;
void dfs(int t,int num,int n,int k){
if(num==k){
vector<int>::iterator it;
cout<<"Vector:";
for(it=v.begin();it!=v.end();it++)
cout<<*it<<" ";
cout<<endl;
ans.push_back(v);
return;
}
for(int i=t;i<=n;i++){
v.push_back(i);
printf("%d %d\n",t,i);
dfs(1+t,num+1,n,k);
v.pop_back();
}
}
vector<vector<int> > combine(int n, int k) {
if(n==0||k==0)
return ans;
dfs(1,0,n,k);
return ans;
}
int main()
{
combine(3,2);
return 0;
}
运行结果如下:
如果把dfs(1+t,num+1,n,k);
改为dfs(++t,num+1,n,k);
那么就可以AC了!
测试代码运行结果为:
这样对比就可以很清楚的看到 ,t+1这样,在递归返回时,并没有改变t本身的值,
这样就会有很多重复的递归,不仅会超时,结果也不对。