描述:
找出所有相加之和为 n 的 k 个数的组合。组合中只允许含有 1 - 9 的正整数,并且每种组合中不存在重复的数字。
说明:
所有数字都是正整数。
解集不能包含重复的组合。
示例 1:
输入: k = 3, n = 7
输出: [[1,2,4]]
示例 2:
输入: k = 3, n = 9
输出: [[1,2,6], [1,3,5], [2,3,4]]
思路分析:
1、如何提高支剪程度来避免不可能有解的无用回溯?这是个问题,值的思考
代码实现:
class Solution {
public List<List<Integer>> ans = new ArrayList<List<Integer>>();
public List<Integer> path = new ArrayList<>();
public void robot(int idx, int k, int n) {
int curSum = 0;
for (int i = 0; i < path.size(); i++) {
curSum += path.get(i);
}
if (idx * (k - 1) > (n - curSum)) return; //支剪
if(idx >= 10 || k <= 0) {
if (k == 0) {
int sum = 0;
List<Integer> temp = new ArrayList<>();
for (int i = 0; i < path.size(); i++) {
temp.add(path.get(i));
sum += path.get(i);
}
if (sum == n) {
ans.add(temp);
}
return;
}
return;
}
path.add(idx);
robot(idx + 1, k - 1, n);
path.remove(path.size() - 1);
robot(idx + 1, k, n);
}
public List<List<Integer>> combinationSum3(int k, int n) {
ans.clear();
if (n > 45) return ans;
if (k == 1 && n > 10) return ans;
robot(1, k, n);
return ans;
}
}
执行用时 :3 ms, 在所有 Java 提交中击败了27.48%的用户
内存消耗 :34.6 MB, 在所有 Java 提交中击败了28.52%的用户