leetcode刷题(20)——538. 把二叉搜索树转换为累加树

一、题目

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

例如:
leetcode刷题(20)——538. 把二叉搜索树转换为累加树_第1张图片

二、思路——递归

累加树,是将每个节点的值改为原来的节点值加上所有大于它的节点值之和。而二叉搜索树的中序遍历的结果是从小到大排列的,比如,给定一个二叉搜索树如下:
leetcode刷题(20)——538. 把二叉搜索树转换为累加树_第2张图片
这棵树中序遍历的结果为

2,3,4,5,6,7,8

由中序遍历结果我们很容易看到,一个节点的右边都是大于它的节点。若将中序遍历的结果反序,则得到

8,7,6,5,4,3,2

也就是反序中序遍历(右子树—>根节点—>左子树)。此时,对于每一个节点来说,大于它的节点在它的左边。因此,使用反序中序遍历时,就可以在遍历每个节点之前,先遍历所有大于它的节点。将这些节点记录并累加至当前节点,就可完成累加树。

由 8,7,6,5,4,3,2 得到累加树的过程如下:

  • 8,7,6,5,4,3,2
  • 8,15(8+7),6,5,4,3,2
  • 8,15,21(15+6),5,4,3,2
  • 8,15,21,26(21+5),4,3,2
  • 8,15,21,26,30(26+4),3,2
  • 8,15,21,26,30,33(30+3),2
  • 8,15,21,26,30,33,35(33+2)

由原二叉搜索树得到的累加树为
leetcode刷题(20)——538. 把二叉搜索树转换为累加树_第3张图片

三、代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    int sum = 0;
    public TreeNode convertBST(TreeNode root) {
        if(root != null){
            convertBST(root.right);
            sum += root.val;
            root.val = sum;
            convertBST(root.left);
        }
        return root;
    }
}

你可能感兴趣的:(leetcode刷题,二叉树,leetcode)