题目:
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
示例 1:
输入:n = 4, k = 2
输出:
[ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4],]
示例 2:
输入:n = 1, k = 1
输出:[[1]]
提示:
1 <= n <= 20
1 <= k <= n
没有剪枝优化
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void dfs(int n, int k, int index)
{
if(path.size()==k){
result.push_back(path);
return;
}
for(int i=index; i<=n; ++i){
path.push_back(i);
dfs(n, k, i+1);
path.pop_back();
}
return ;
}
vector<vector<int>> combine(int n, int k) {
dfs(n, k, 1);
return result;
}
};
剪枝优化
class Solution {
public:
vector<vector<int>> result;
vector<int> path;
void dfs(int n, int k, int index)
{
if(path.size()==k){
result.push_back(path);
return;
}
for(int i=index; i<=n-(k-path.size()) + 1; ++i){ //剪枝优化
path.push_back(i);
dfs(n, k, i+1);
path.pop_back();
}
return ;
}
vector<vector<int>> combine(int n, int k) {
dfs(n, k, 1);
return result;
}
};