剑指 Offer 34. 二叉树中和为某一值的路径
题目描述
输入一棵二叉树和一个整数,打印出二叉树中节点值的和为输入整数的所有路径。从树的根节点开始往下一直到叶节点所经过的节点形成一条路径。
示例
给定如下二叉树,以及目标和 target = 22,
返回:
提示
节点总数 <= 10000
思路
注意这里的路径是指:一直从根节点一直走到最后的节点(该节点没有子节点)举个例子:以上面的树为例 target的值为26时 路径为 [[5,8,13]] 但target 为 17时 明明可以走:5->8->4 但是 题目的输出路径为 []
一切按题目来:
- 用一个 result 数组保存路径
- 用一个 temp 数组用来选择节点
- 选择节点后 判断节点值是否为 target 值且没有子节点 是:说明到达这个节点就是一条路径的终点 否:递归调用函数 将节点的子节点做参数传入继续搜索 (因为加入了当前节点 要减去当前节点的值 更新 target值)
- 把当前节点 退出数组 选择下一节点
代码
/**
* Definition for a binary tree node.
* function TreeNode(val, left, right) {
* this.val = (val===undefined ? 0 : val)
* this.left = (left===undefined ? null : left)
* this.right = (right===undefined ? null : right)
* }
*/
/**
* @param {TreeNode} root
* @param {number} target
* @return {number[][]}
*/
var pathSum = function(root, target) {
if(!root) return [];
//结果数组
let result = [];
dfs(root, result, target, []);
return result;
};
let dfs = function(node, result, target, temp){
if(!node) return;
//选择节点
temp.push(node.val);
//当前节点刚刚好等于 更新后的target值 路径终点
if(!node.left && !node.right && target === node.val){
result.push([...temp]);
}else{
//当前节点不是终点 更新target值 继续搜索
dfs(node.left, result, target - node.val, temp);
dfs(node.right, result, target - node.val, temp);
}
temp.pop(node);
}