655. 輸出二叉樹

題目:
在這裏插入圖片描述在這裏插入圖片描述
題解:
思路: 我們發現輸出的數組的行數就是樹的高度,列數就是樹的高度的冪次方-1.

  1. 求出樹的高度
  2. 利用二分法將節點值插入。

代碼:

var printTree = function (root) {
    if (root === null) {
        return [];
    }
    let n = treeTop(root);
    let col = Math.pow(2, n) - 1;
    var res = [];
    for (let i = 0; i < n; i++) {
        let resSub = []
        for (let j = 0; j < col; j++) {
            resSub.push('')
        }
        res.push(resSub)
    }

    getRow(res, root, 0, 0, col)

    function getRow(res, root, row, left, right) {
        if (root != null) {
            let mid = parseInt((left + right) / 2);
            res[row][mid] = (root.val).toString()
            getRow(res, root.left, row + 1, left, mid)
            getRow(res, root.right, row + 1, mid + 1, right)
        }
    }

    function treeTop(root) {
        if (root === null) {
            return 0
        } else {
            return 1 + Math.max(treeTop(root.left), treeTop(root.right))
        }
    }

    return res;
};

注:(這裏手動標紅)

在這裏插入圖片描述

(ps:第二週,第三天)

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章