【LeetCode】第538题：把二叉搜索树转换为累加树（百度面试题）

博主秋招提前批已拿百度、字节跳动、拼多多、顺丰等公司的offer，可加微信：pcwl_Java 一起交流秋招面试经验，可获得博主的秋招简历和复习笔记。

LeetCode 链接：https://leetcode-cn.com/problems/convert-bst-to-greater-tree/submissions/

题目：给定一个二叉搜索树（Binary Search Tree），把它转换成为累加树（Greater Tree)，使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

例如：

分析：

根据二叉搜索树的特性可知：二叉搜索树的中序遍历是一个递增的数组，那么只需要在中序遍历数组将一个数右边的数累加起来，再加到其自身身上就可以了。为了实现这个过程，我们可以采取二叉搜索树中序遍历的逆序过程，即：右 --> 中 --> 左 的遍历模式，这样就可以将右边的数先收集起来进行累加，遍历到哪个数只需要将当前累加的 count 加上自身的值就是自己在累加树中的值，然后更新 count（需要加上自身的值，即此时 count 就为累加树中当前节点的值）。

代码如下：

public class ConvertBST_538 {

    class TreeNode{
        int val;
        TreeNode left;
        TreeNode right;

        public TreeNode(int val){
            this.val = val;
        }
    }

    int count = 0;  // 累加和

    public TreeNode convertBST(TreeNode root) {
        if(root == null){
            return root;
        }

        // 逆序遍历中序遍历实现：右 --> 中 --> 左
        convertBST(root.right);
        // 将当前节点的中序遍历的右边节点都累加起来加到自身上面
        root.val += count;
        count = root.val;
        // 遍历左子树
        convertBST(root.left);
        return root;
    }
}