JavaScript|LeetCode|搜索|257.二叉树的所有路径

二叉树路径遍历算法

最新推荐文章于 2022-02-16 16:33:41 发布

原创最新推荐文章于 2022-02-16 16:33:41 发布 · 299 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #js #算法

LeetCode 专栏收录该内容

43 篇文章

订阅专栏

本文介绍两种求解二叉树所有路径的算法：递归法与非递归法。递归法通过分解问题，将大问题转化为子问题解决；非递归法采用队列实现按层扫描，存储节点及其路径，直至找到叶子节点。

法1：递归
想法：

大问题包含小问题，大小问题是同种问题
找到root的左子树和右子树的所有可能路径（从左子树的根到叶子，从右子树的根到叶子）（分为left和right两类），分别与root衔接，得到最终结果
找左子树的所有可能的路径（从左子树的根到叶子）（此时左子树的根为新的root’）：找到root’的左子树和右子树的所有可能路径（分为left‘和right’两类），分别与root‘衔接，得到最终结果。返回该结果，则是原root的左子树的所有可能路径
右子树同理

/** 
* 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];    
    }    
    var temp = root.val + "->";    
    var output = [], left = binaryTreePaths(root.left),                        
        right = binaryTreePaths(root.right);    
    var i = 0;    
    for(i = 0; i < left.length; i++) {        
        output[output.length] = temp + left[i];    
    }    
    for(i = 0; i < right.length; i++) {        
        output[output.length] = temp +  right[i];    
    }    return output;
};

法2：非递归
看了答案，打算将非递归的方法练习一下
想法：

按层扫描，用队列存储当前层的节点和节点对应的路径（以该节点为最后一个节点的路径）
从队列中取出节点，若该节点为叶子节点，则对应的路径为答案之一
队列中存储数组[Treenode, 以Treenode为最后一个节点的路径]

/** 
* 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];    
    }    
    var output = [], array = [];     
    // array存储数组，数组元素为[Treenode, 以Treenode为最后一个节点的路径]        
    
    var length = 0; // 存储当前层root节点的个数（如根节点的左右子树的根节点）    
    array[array.length] = [root, "" + root.val];
    while(array.length > 0) {        
        length = array.length;        
        while(length > 0) {            
            if(array[0][0].left == null && array[0][0].right == null) { // 叶子节点                
                output[output.length] = array[0][1];                
                array.shift();                
                length--;                
                continue;            
            }            
            if(array[0][0].left != null) {                
                array.push([array[0][0].left, array[0][1] + "->" + array[0][0].left.val]);            
            }            
            if(array[0][0].right != null) {                
                array.push([array[0][0].right, array[0][1] + "->" + array[0][0].right.val]);            
            }            
            array.shift();            
            length--;        
        }    
    }    
    return output;
};