【剑指offer】二叉树中和为某一值的路径(JS)

题目描述:

输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)

思路:

看到这个题,首先想到深搜,深度遍历直到叶子节点,不符合条件再出栈回到上一层节点,注意结束条件以及最后的回溯

/* function TreeNode(x) {
    this.val = x;
    this.left = null;
    this.right = null;
} */
function FindPath(root, expectNumber)
{
    // write code here
   let result = [], path = [], cur = 0;
    if(!root) return result;
    dfs(root, path, result, cur, expectNumber);
    return result;
}
function dfs(root, path, result, cur, exp) {
    cur += root.val;
    path.push(root.val);
    //如果符合条件则将路径保存起来
    if(cur == exp && root.left == null && root.right == null){
        result.push(path.slice(0));
    }
    if(root.left){
        dfs(root.left, path, result, cur, exp);
    }
    if(root.right){
        dfs(root.right, path, result, cur, exp);
    }
    path.pop();
}

你可能感兴趣的:(剑指offer)