【leetcode】538. 把二叉搜索树转换为累加树

🚀题目

leetcode原题链接

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。

示例 1：

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

示例 2：
输入：root = [0,null,1]
输出：[1,null,1]

示例 3：
输入：root = [1,0,2]
输出：[3,3,2]

示例 4：
输入：root = [3,2,4,1]
输出：[7,9,4,10]

提示：
树中的节点数介于 0 和 104 之间。
每个节点的值介于 -104 和 104 之间。
树中的所有值 互不相同 。
给定的树为二叉搜索树。

💥leetcode代码模板

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {TreeNode}
 */
var convertBST = function(root) {

};

🚀思路

🎲方法一：递归

🟡首先来看一下二叉树是如何累加的，如下图所示，是按照右中左的顺序累加每一个节点的值。因此我们可以利用递归来进行**“右中左”**顺序的遍历。

🟢OK,遍历的方式已经解决了，那么如何来修改节点的值和存储累加和呢？我们可以初始化一个全局的变量sum = 0，遇到一个节点时就更新sum即sum += node.val，然后修改节点的值node.val = sum

🔴递归三部曲：
1️⃣递归函数的参数和返回值：参数就是二叉树的节点，不需要返回值，只需要在遍历的过程中修改节点值。

2️⃣递归的返回条件是：遇到空节点时什么也不做，直接返回。

3️⃣单层递归的逻辑：按照右中左的顺序进行递归，即

traverse(node.right)
//处理中间节点
...
traverse(node.left)

对于中间节点（即当前节点），首先更新sum即sum += node.val，然后更新节点值node.val = sum

🎲方法二：迭代

把上面递归改为迭代，可以使用二叉树迭代遍历的统一模板【leetcode】二叉树遍历迭代法的统一模板。

💻代码

🎲方法一：递归

var convertBST = function(root) {
    function traverse(node){
        if(!node) return 
        traverse(node.right)
        sum += node.val
        node.val = sum
        traverse(node.left)
    }

    let sum = 0
    traverse(root)
    return root
};

方法二：迭代

var convertBST = function(root) {
    let sum = 0
    let stack = []
    if(root) stack.push(root)
    while(stack.length){
        // 当前指针指向栈顶节点
        let cur = stack[stack.length - 1]

        if(cur){    // 当前节点不为空
            stack.pop()
            if(cur.left) stack.push(cur.left)
            stack.push(cur)
            stack.push(null)
            if(cur.right) stack.push(cur.right)
        }else{  //当前节点为空
            // 弹出空节点
            stack.pop()
            // 弹出中间节点
            let node = stack.pop()
            sum += node.val
            node.val = sum
        }
    }
    return root
};

🍪总结

这道题目不难，通过观察累加的顺序发现是要用右左中的顺序进行遍历，遍历过程中借助一个变量来累加和更新节点值就好。

✨$我是coner，请别关注我的专栏，本系列文章为个人刷题记录（偷偷自己卷🤤）：$leetcode题解js

✨$每天早上更新3道leetcode题目的js题解🚀🚀🚀$