257. 二叉树的所有路径(JavaScript)
给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
输入:
1
/ \
2 3
\
5
输出: ["1->2->5", "1->3"]
解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3
思路:
使用递归方法来解决此问题。对于根节点,如果根节点为空,则返回空数组[],如果为叶子节点,则返回包含此节点的值的数组(这个数组只有一个元素)。若不为空也不是叶子节点,则对左右子树递归调用,将两个结果数组拼接起来,最后使用 map 函数来对数组的每个元素进行字符串的操作。
/**
* Definition for a binary tree node.
* function TreeNode(val) {
* this.val = val;
* this.left = this.right = null;
* }
*/
/**
* @param {TreeNode} root
* @return {string[]}
*/
var binaryTreePaths = function(root) {
if (root === null) return [];
if (root.left === null && root.right === null) {
return [root.val.toString()];
}
var left = binaryTreePaths(root.left),
right = binaryTreePaths(root.right);
return left.concat(right).map(x => root.val + '->' + x);
};