题目:77. 组合
给定两个整数 n 和 k,返回 1 … n 中所有可能的 k 个数的组合。
示例:
输入: n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/combinations
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解
1 2 3 4
/ | \ / \ /
2 3 4 3 4 4
dfs(1,4,2)
代码
class Solution {
vector<vector<int>> ans;
vector<int> cmb;
public:
vector<vector<int>> combine(int n, int k) {
dfs(1, n, k);
return ans;
}
void dfs(int cur, int n, int k) {
// 剪枝:不可能构造出长度为k的组合时
if (cmb.size() + (n - cur + 1) < k) {
return;
}
// 满足条件时保存结果
if (cmb.size() == k) {
ans.push_back(cmb);
return;
}
// 结束条件
if (cur == n + 1) {
return;
}
//选择cur
cmb.push_back(cur);
dfs(cur + 1, n, k);
//回溯
cmb.pop_back();
// 不选择cur
dfs(cur + 1, n, k);
}
};