LeetCode 216:简单思路解决---DFS

mac2026-09-29  5

一、先祭模板:

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解答思路,也可能有其他简单的解法,有大佬知道的也可以在下面留言讨论。

 

最新回复(0)