// 算法一
public List<List<Integer>> combine(int n, int k) {
List<Integer> temp = new ArrayList<>();
List<List<Integer>> result = new ArrayList<>();
bakcTrack(temp, result, 1, n, k);
return result;
}
public void bakcTrack(List<Integer> temp, List<List<Integer>> result, int index, int n, int k) {
if (temp.size() == k) {
result.add(new ArrayList<>(temp));
return;
}
for (int i = index; i <= n; i++) {
temp.add(i);
bakcTrack(temp, result, i + 1, n, k);
temp.remove(temp.size() - 1);
}
}
// 算法二
List<Integer> temp = new ArrayList<Integer>();
List<List<Integer>> ans = new ArrayList<List<Integer>>();
public List<List<Integer>> combine(int n, int k) {
dfs(1, n, k);
return ans;
}
public void dfs(int cur, int n, int k) {
// 剪枝:temp 长度加上区间 [cur, n] 的长度小于 k,不可能构造出长度为 k 的 temp
if (temp.size() + (n - cur + 1) < k) {
return;
}
// 记录合法的答案
if (temp.size() == k) {
ans.add(new ArrayList<Integer>(temp));
return;
}
// 考虑选择当前位置
temp.add(cur);
dfs(cur + 1, n, k);
// 考虑不选择当前位置
temp.remove(temp.size() - 1);
dfs(cur + 1, n, k);
}
组合算法实现
于 2022-12-04 00:26:56 首次发布