题目
给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
你可以按 任何顺序 返回答案。
解题思路
递归+回溯。注意剪枝的情况,即已经拿的数字个数 + 剩下数字个数 < k
的时候,后面的就不用判断了。
代码
class Solution {
public List<List<Integer>> combine(int n, int k) {
List<List<Integer>> ans = new ArrayList<>();
dfs(ans, new ArrayList<Integer>(), 1, n, k);
return ans;
}
private void dfs(List<List<Integer>> ans, ArrayList<Integer> now, int pos, int n, int k) {
if (now.size() == k) {
ans.add(now);
return;
}
for (int i = pos; i <= n; i++) {
if (now.size() + n - i + 1 < k) break;
now.add(i);
dfs(ans, new ArrayList<Integer>(now), i + 1, n, k);
now.remove(now.size() - 1);
}
}
}