给定两个整数 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中最后一个元素}}