前端算法-二叉树的全部路径

688 阅读2分钟

开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第36天,点击查看活动详情

题目

给你一个二叉树的根节点 root ,按 任意顺序 ,返回所有从根节点到叶子节点的路径。

输入: root = [1,2,3,null,5]
输出: ["1->2->5","1->3"]

思路一

我们先新建一个result变量,它是一个数组,然后我们在对形参root做判断没,如果形参root为null,则直接将result空数组进行返回,在创建一个list变量,它也是一个空数组,然后我们呢将root的val属性值使用push方法添加到list数组中,然后再声明一个helper函数,它是一个递归函数,用于递归遍历子节点,它接受三个参数,第一个形参是返回结果的数组,第二个形参是用于存放当前遍历值的数组,第三个形参数是我们需要遍历的数组,然后我们将result数组和list数组以及root形参传递进去,在helper函数中,我们先判断root形参的左节点和右节点是否全部都为null,如果是我们就声明一个sb变量,它是一个空字符串,然后进行循环list数组,在循环中我们将list数组的当前值和'->'字符使用concat方法进行拼接起来,最后在将list数组的最后一个值拼接到sb字符串的末尾,在使用push方法添加到result数组中,最后return出去即可,这里也是递归终止条件,如果形参root的左节点和右节点都不为null,则先判断当前的形参root的左节点是否不为null如果不为null,我们就将形参root左节点的val参数使用push方法添加到list数组中,然后在调用自身,只不过把当前第三个形参换成了形参root的左节点,最后用pop方法把list数组中最末尾的值删除掉即可,右节点也是如此,最后我们将result数组返回出去

var binaryTreePaths = function(root) {
    let result = []
    if (root == null) {
        return result
    }
    let list = []
    list.push(root.val)
    helper(result, list, root)
    return result
};
let helper = function(result, list, root) {
    if (root.left == null && root.right == null) {
        let sb = ''
        for (let i = 0; i < list.length - 1; i++) {
            sb = sb.concat(list[i]).concat('->')
        }
        sb = sb.concat(list[list.length - 1])
        result.push(sb)
        return
    }
    if (root.left != null) {
        list.push(root.left.val)
        helper(result, list, root.left)
        list.pop()
    }
    if (root.right != null) {
        list.push(root.right.val)
        helper(result, list, root.right)
        list.pop()
    }
}