【LeetCode】538. 把二叉搜索树转换为累加树（中等）——代码随想录算法训练营Day23

题目链接：538. 把二叉搜索树转换为累加树

题目描述

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

提醒一下，二叉搜索树满足下列约束条件：

• 节点的左子树仅包含键 小于 节点键的节点。
• 节点的右子树仅包含键 大于 节点键的节点。
• 左右子树也必须是二叉搜索树。

注意：本题和 1038: 力扣（LeetCode）官网 - 全球极客挚爱的技术成长平台 相同

示例 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：538.把二叉搜索树转换为累加树_哔哩哔哩_bilibili

题解1：递归法

思路：以右中左的顺序进行中序遍历，用另一个指针记住上一个节点的值，将当前节点的值加上上个节点的值，即完成累加树转换。

/**
 * 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) {
    let pre = null;
    const inorder = function (node) {
        if (!node) {
            return;
        }
        inorder(node.right); // 右
        // 中
        if (pre) {
            node.val += pre.val;
        }
        pre = node;
        inorder(node.left); // 左
    }
    inorder(root);
    return root;
};

分析：时间复杂度为 O(n)，空间复杂度为 O(logn)。

题解2：迭代法

思路：中序遍历的迭代法写法。

/**
 * 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) {
    let pre = null, cur = root;
    const stack = [];
    while (stack.length > 0 || cur) {
        if (cur) {
            stack.push(cur);
            cur = cur.right; // 右
        } else {
            cur = stack.pop();
            // 中
            if (pre) {
                cur.val += pre.val;
            }
            pre = cur;
            cur = cur.left; // 左
        }
    }
    return root;
};

分析：时间复杂度为 O(n)，空间复杂度为 O(logn)。

收获

利用二叉搜索树的特性，使用中序遍历解决此题。