216.组合总和III
找出所有相加之和为 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]]
思路:
使用回溯算法,
LinkedList path: 来存储路径
List<List> res: 存储结果
backtrack(int n,int k,int start): 递归函数
path.size()k && sumn: 单层递归结束条件,也是res收集结果条件
sum>n || path.size(): 当path.size()大于k或者sum>n时,就直接return了,已经没有必要再递归下去了。
单层递归逻辑:
for (int j = start; j <= n-(k-path.size())+1; j++) {
path.add(j);
backtrack(n,k,j+1);
path.removeLast();
}
注意:由于是组合问题,不是排列问题,所以使用start来移动元素。
import java.util.ArrayList;
import java.util.Arrays;
import java.util.LinkedList;
import java.util.List;
/**
* 216.组合总和III
* 力扣题目
* 找出所有相加之和为 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]]
*/
public class Combination3 {
List<List<Integer>> res = new ArrayList<>();
LinkedList<Integer> path = new LinkedList<>();
int sum = 0;
public List<List<Integer>> combinationSum3(int k, int n) {
backtrack(k,n,1);
return res;
}
private void backtrack(int k,int n,int start){
if (sum>n || path.size()>k) return;
if (path.size()==k && sum==n){
res.add(new ArrayList<>(path));
return;
}
for (int i = start; i <= 9; i++) {
path.add(i);
sum+=i;
backtrack(k,n,i+1);
sum-=i;
path.removeLast();
}
}
}