LeetCode 113. 路径总和 II
作者:Choi Yang
更新于:7 个月前
字数统计:426 字
阅读时长:1 分钟
题目描述
给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
javascript
给定如下二叉树,以及目标和 sum = 22,
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回:
javascript
[
[5, 4, 11, 2],
[5, 8, 4, 5],
];
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/path-sum-ii 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路
dfs
,进行深度优先遍历,一直遍历到子节点为止,进行一次判断,如果当前 sum
为 0 ,那么就是我们想要的结果,然后注意 js
语法中形参如果是数组,那么我们拿到的是引用值,可以拷贝一份。
javascript
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @param {number} sum
* @return {number[][]}
*/
var pathSum = function (root, sum) {
if (!root) return [];
let res = [];
let dfs = (cur, root, sum) => {
if (root == null) return 0;
// 拷贝一份
cur = [...cur, root.val];
sum -= root.val;
if (!root.left && !root.right && sum == 0) {
res.push(cur);
return;
}
// 优先遍历左子树
root.left && dfs(cur, root.left, sum);
root.right && dfs(cur, root.right, sum);
};
dfs([], root, sum);
return res;
};
不太明白的小伙伴,这里给一个友好的提示,我们可以打印一下拷贝出来的cur
,结合图示应该就好理解了,经典的 dfs
实现的先序遍历。
javascript
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
javascript
[5][(5, 4)][(5, 4, 11)][(5, 4, 11, 7)][(5, 4, 11, 2)][(5, 8)][(5, 8, 13)][
(5, 8, 4)
][(5, 8, 4, 5)][(5, 8, 4, 1)];
javascript
学如逆水行舟,不进则退