第十七题 剑指 Offer 34. 二叉树中和为某一值的路径

剑指 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);
}