leetcode 998.最大二叉树 II typescript/javascript

leetcode-998 最大二叉树 II typescript/javascript

题目

解题思路

根节点+右子树遍历

看听雨的评论搞出来了，原本没读懂题目，大概就是在 a（例子中的） 中选最大的数作为根节点，该数左边的为左子树,右边的为右子树,然后左边的数选最大作为左节点，右边的数选最大作为右节点，然后在左右节点分别执行同样的操作。因为 val 是最后添加到 a 也就是形成了 b，它一直是在右边所以只需要遍历右子树找到插入的位置就好了。如果它比 a 中的所有数都大，那么直接把 val 作为根节点。

代码

/**
 * Definition for a binary tree node.
 * class TreeNode {
 *     val: number
 *     left: TreeNode | null
 *     right: TreeNode | null
 *     constructor(val?: number, left?: TreeNode | null, right?: TreeNode | null) {
 *         this.val = (val===undefined ? 0 : val)
 *         this.left = (left===undefined ? null : left)
 *         this.right = (right===undefined ? null : right)
 *     }
 * }
 */

function insertIntoMaxTree(root: TreeNode | null, val: number): TreeNode | null {
    if(val > root.val){
        let rootTree:TreeNode = new TreeNode(val,root,null);
        return rootTree;  
    }else{
        let pre:TreeNode = root;
        while(pre.right && val<pre.right.val){
            pre = pre.right;
        }
        if(pre.right){
            pre.right = new TreeNode(val,pre.right,null);
        }else{
            pre.right = new TreeNode(val,null,null);
        }
        return root;
    }
};