给定两个整数 n
和 k
,返回范围 [1, n]
中所有可能的 k
个数的组合。
未剪枝版:
public List<List<Integer>> combine(int n, int k) {
backtracking(n,k,1);
System.out.println(res);
return res;
}
public void backtracking(int n,int k,int start){
if(path.size()==k){
//直接 res.add(path),会导致 res 中保存的是 path 的引用,而不是它的当前状态的副本。
// 这样做的话,当 path 发生变化时,res 中所有保存的路径也会跟着改变。
res.add(new LinkedList<>(path));
return;
}
for(int i=start;i<=n;i++){
path.add(i);
backtracking(n,k,i+1);
path.removeLast(); //回溯,删除path中最后一个元素
}
}
剪枝版:
剪枝条件: 剩余的元素大于 还需要的元素
剩余元素:n-i+1
还需元素:k-path.size()
n-i+1>=k-path.size(), 可得 i<=n-(k-path.size)+1
public void backtracking(int n,int k,int start){
if(path.size()==k){
//直接 res.add(path),会导致 res 中保存的是 path 的引用,而不是它的当前状态的副本。
// 这样做的话,当 path 发生变化时,res 中所有保存的路径也会跟着改变。
res.add(new LinkedList<>(path));
return;
}
//剪枝条件: 剩余的元素大于 还需要的元素
//剩余元素:n-i, 还需元素个数:k-path.size()
//n-i>=k-path.size(),可得 i<=n-(k-path.size)
for(int i=start;i<=n-(k-path.size());i++){
path.add(i);
backtracking(n,k,i+1);
path.removeLast(); //回溯,删除path中最后一个元素
}
}