LeetCode 538. 把二叉搜索树转换为累加树

目录结构

1.题目

2.题解


1.题目

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

示例:

输入: 原始二叉搜索树:
              5
            /   \
           2     13

输出: 转换为累加树:
             18
            /   \
          20     13

注意:本题和 1038: https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree/相同

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

2.题解

反向中序遍历。

即先右子树,然后根,最后左子树。

public class Solution538 {

    @Test
    public void test538(){
        TreeNode root = new TreeNode(5);
        root.left = new TreeNode(2);
        root.right = new TreeNode(13);
        TreeNode.printf(convertBST(root));
    }

    public TreeNode convertBST(TreeNode root) {
        inorder(root);
        return root;
    }

    private int sum = 0;

    public void inorder(TreeNode root){
        if (root != null){
            inorder(root.right);
            sum += root.val;
            root.val = sum;
            inorder(root.left);
        }
    }
}
  • 时间复杂度:O(n)
  • 空间复杂度:O(n)

你可能感兴趣的:(LeetCode)