538. 把二叉搜索树转换为累加树 09-21

题目

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

例如:

输入: 原始二叉搜索树:
5
/
2 13
输出: 转换为累加树:
18
/
20 13

关于二叉搜索树

开始看题目的时候,没注意二叉搜索树的定义

导致第一次实现代码如下:
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public TreeNode convertBST(TreeNode root) {
        dfs(root);
        return root;
    }

    public void dfs(TreeNode root){
错误认为比较当前递归三个节点的值的大小,然后找大值求和重新赋值即可
        int roo=0,left=0,right=0;
        if(root==null){
            return;
        }
        if(root.left!=null){
            if(root.left.val>root.val){
                roo+=root.left.val;
            }
            else{
                left+=root.val;
            }
            if(root.right!=null){
                if(root.right.val>root.left.val){
                    left+=root.right.val;
                }
                else{
                    right+=root.left.val;
                }
            }   
        }

        if(root.right!=null){
            if(root.right.val>root.val){
                roo+=root.right.val;
            }
            else{
                right+=root.val;
            }     
        }
        root.val+=roo;
        if(root.left!=null){
            root.left.val+=left;
        }
        if(root.right!=null){
            root.right.val+=right;
        }
        dfs(root.left);
        dfs(root.right);
    }
}

上述代码实际测试时,[2,0,3,-4,1]会出现第三层的右叶子节点无法加二层比其大的值的和
事实上,二叉搜索树定义如下
二叉搜索树是一棵空树,或者是具有下列性质的二叉树:
1. 若它的左子树不空,则左子树上所有节点的值均小于它的根节点的值;
2. 若它的右子树不空,则右子树上所有节点的值均大于它的根节点的值;
3. 它的左、右子树也分别为二叉搜索树。
由这样的性质我们可以发现,二叉搜索树的中序遍历是一个单调递增的有序序列。如果我们反序地中序遍历该二叉搜索树,即可得到一个单调递减的有序序列。

因此,正确代码如下:

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

    public void dfs(TreeNode root){
        if(root==null){
            return;
        }        
        dfs(root.right);//反中序遍历,即值递减排列
        root.val+=roo;//roo需要为全局变量,本层递归root.val与比自己大的节点值求和
        roo=root.val;//roo记录本次的大节点值供下次使用,
//本次大节点值已经求过和,下次直接加roo即可,即root.val+=roo;
        dfs(root.left);
    }
}

你可能感兴趣的:(538. 把二叉搜索树转换为累加树 09-21)