题目描述
给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例
给定如下二叉树，以及目标和 sum = 22
5
/ \
4 8
/ / \
11 13 4
/ \ / \
7 2 5 1
返回
[
[5,4,11,2],
[5,8,4,5]
]题目分析
其实这道题跟路径总和这道题很相似
- 其实就是将每次遍历的数放到
prefix中，判断当前 节点是否有子节点 还有sum - value === 0, 是的话，放入到res - 有 左节点 进入递归
- 有 右节点 进入递归
实现
/**
* 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) {
let res = [];
let prefix = []
if(!root) {
return res
}
dfs(root, sum, res, prefix);
return res;
};
var dfs = function(root, sum, res, prefix) {
const value = root.val;
const remain = sum - value;
// 如果叶子节点都没有而且 sum - value == 0, 那么我们将prefix 加进数组
if(!root.left && !root.right && remain === 0) {
res.push([...prefix, value])
return ;
}
// 没有子节点 且 remain !== 0 那就停止递归
if(!root.left && !root.right && remain !== 0) {
return;
}
// 有左子节点
if(root.left) {
dfs(root.left, remain, res, [...prefix, value])
}
// 有右子节点, 进入递归
if(root.right) {
dfs(root.right, remain, res, [...prefix, value]);
}
}总结
像这种需要保留 状态集的可以用prefix 装着，最后当符合条件的时候再 把 prefix 装到 result
