打印二叉树递归函数

时间:2019-04-19 10:22:35

标签: javascript binary-tree

我正在尝试返回一个表示二叉树的数组。我创建了一个输出数组,其中填充了空字符串数组,其中每个数组代表树的一个级别,字符串代表该级别上每个可能的节点位置。由于某种原因,我的递归函数似乎正在更改父输出数组中的所有数组,而不仅仅是适当的数组。

var printTree = function(root) {
//first find depth of tree
    let depth = 0
    const findDepth = (node, level) => {
        depth = Math.max(depth, level);
        if (node.left) {
            findDepth(node.left, level + 1)
        }
        if (node.right) {
            findDepth(node.right, level + 1)
        }
    }
    findDepth(root, 1);
    let width = 1 + ((depth - 1) * 2)
//create array of arrays filled with blanks that match height and width
// of given tree
    let output = new Array(depth).fill(new Array(width).fill(''));
    let mid = Math.floor(width / 2);
//do DFS through tree and change output array based on position in tree
    const populate = (node, level, hori) => {
        output[level][hori] = node.val;
        if (node.left) {
            populate(node.left, level + 1, hori - 1);
        }
        if (node.right) {
            populate(node.right, level + 1, hori + 1);
        }
    }
    populate(root, 0, mid);
    return output;
};

如果我将一棵二叉树的根节点的val设为1,则其唯一的子节点的val设为2。

我的输出数组应该是:

[['', 1 , ''],
[2 , '' , '']]

但是它看起来像这样:

[[2, 1, ''],
[2, 1, '']]

我已经在控制台上记录了递归调用,但我想不出为什么要在矩阵的所有行中进行这些更改,而不仅仅是在适当的级别上进行

我该如何解决这个问题?

1 个答案:

答案 0 :(得分:1)

您需要更改此行

let output = new Array(depth).fill(new Array(width).fill(''));
//                                 ^^^^^^^^^^^^^^^^^^^^^^^^^ same array!

进入

let output = Array.from({ length: depth }, _ => Array.from({ length: width }).fill(''));

因为您用相同的数组填充了数组。带下划线的部分填充相同的数组,即一个常数值。