这道题目可以理解为,input是一个包含1-9的array,求出所有长度为k的,能够sum up成n的组合。每个数字在每个组合里只可以用一次,组合不可以重复。感觉像是combination和combination sum的合体。我还是打算用dfs做,那么难点就在于base case 和 recursion。
感觉这类题目都是在input的基础上递归,寻找以input里某个数字开头的结果。所以这道题目的代码其实和类似题差别不大,只需要自行添加input,增加检查结果长度的代码即可。
public class Solution {
public List<List<Integer>> combinationSum3(int k, int n) {
List<List<Integer>> result = new ArrayList<List<Integer>>();
if (k < 0 || k > 9 || n <= 0) {
return result;
}
int[] nums = new int[]{1, 2, 3, 4, 5, 6, 7, 8, 9};
helper(result, new ArrayList<Integer>(), k, n, nums, 0);
return result;
}
public void helper(List<List<Integer>> result, List<Integer> sub, int k, int n, int[] nums, int start)
{
if (sub.size() == k && n == 0) {
result.add(new ArrayList<Integer>(sub));
return;
}
for (int i = start; i < nums.length; i++) {
sub.add(nums[i]);
helper(result, sub, k, n - nums[i], nums, i + 1);
sub.remove(sub.size() - 1);
}
}
}