一、题目
给定两个整数 n 和 k,返回范围 [1, n] 中所有可能的 k 个数的组合。
你可以按 任何顺序 返回答案。
[来源:https://leetcode-cn.com/problems/combinations/]
示例 1:
输入:n = 4, k = 2
输出:
[
[2,4],
[3,4],
[2,3],
[1,2],
[1,3],
[1,4],
]
示例 2:
输入:n = 1, k = 1
输出:[[1]]
二、题解
class Solution {
List<Integer> temp = new ArrayList<>();
List<List<Integer>> result = new ArrayList<>();
public List<List<Integer>> combine(int n, int k) {
dfs(1, n, k);
return result;
}
public void dfs(int curr, int n, int k) {
//剪枝:temp 长度加上区间 [curr, n]的长度小于 k,不可能构造出长度为 k 的 temp
if (temp.size() + (n - curr + 1) < k) {
return;
}
//记录合法答案
if (temp.size() == k) {
result.add(new ArrayList<>(temp));
return;
}
//选择当前位置
temp.add(curr); //向临时链表添加当前元素
dfs(curr + 1, n, k); //处理元素,加入结果集
temp.remove(temp.size() - 1); //处理完毕,移除尾部元素
//不选择当前位置
dfs(curr + 1, n, k);
}
}
本题的递归树(仅列出部分递归树):