一、先祭模板:
void dfs(参数)
{
if(满足结束条件)
{
计数或进行其他操作;
return;
}
for(查找当前节点的周围的节点)
{
进行其他的操作;
标记已经搜索过的节点;
dfs(下一次搜索的节点);
取消标记;
}
}
二、再上例子
LeetCode 216:
找出所有相加之和为 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]]
三、代码
.代码可以看做是完全套上面模板解答的,类似问题还有39T,77T等问题,大家都可以套这个模板解决。
public List<List<Integer>> combinationSum3(int k, int n) {
List<List<Integer>> re = new ArrayList<>();
List<Integer> temp = new ArrayList<>();
DFS(re, temp, k, n, 0, 1);
return re;
}
public static void DFS(List<List<Integer>> re, List<Integer> temp, int k, int n, int sum, int index) {
if (temp.stream().mapToDouble(x -> x).sum() == n && temp.size() == k) {
re.add(new ArrayList<>(temp));
return;
}
for (int i = index; i < 10; i++) {
if (temp.size() < k && sum + i <= n) {
temp.add(i);
DFS(re, temp, k, n, sum + i, i + 1);
temp.remove(temp.size() - 1);
}
}
}
作者认为类似这种题的思路最容易想到的就是dfs解答思路,也可能有其他简单的解法,有大佬知道的也可以在下面留言讨论。