地址 https://leetcode-cn.com/problems/combinations/
给定两个整数 n 和 k,返回 1 ... n 中所有可能的 k 个数的组合。 示例: 输入: n = 4, k = 2 输出: [ [2,4], [3,4], [2,3], [1,2], [1,3], [1,4], ]
解答
观察样例发现
当在某一个位置选中一个数后 后面填充的数字必须比它大
这样可以避免选中 1 5 组合后 又出现了 5 1 组合
使用DFS 完成上述操作
那么 每个数字都有两种可能 填充或者 略过
如果操作的值超过n 那么就不再继续
如果当前已经填充K个数字了 那么放入答案不再继续
class Solution { public: vector<vector<int>> ans; void dfs(int n, int k, int idx, vector<int> v) { if(v.size()>k) return; //剩下的元素 全选也不足以达到选择K个元素 剪枝 if( (v.size() + n-idx+1) < k ) return; if (v.size() == k) {ans.push_back(v);return;} v.push_back(idx); dfs(n, k, idx + 1, v); v.pop_back(); dfs(n, k, idx + 1, v); } vector<vector<int>> combine(int n, int k) { vector<int> v; dfs(n, k, 1, v); return ans; } };