leetcode538(把二叉搜索树转换为累加树:二叉树的中序遍历)

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

例:
输入:
[5,2,13,null,4]
输出:
[18,24,13,null,22]

题解:既然题目要求每个节点的值改成原来节点的值加上所有大于它的节点值之和,所以我们要知道对于每一个节点值,大于它的节点值有哪些,中序遍历就能很好的解决这个问题,中序遍历可以直接得到升序数组,我们只需要将得到的升序数组转换成累加数组(数组中的每个元素是原当前数组元素加上它后面的数组元素),再将累加数组的值带到二叉搜索树替换即可。

class Solution {
    private final ArrayList<Integer>nodeArray=new ArrayList<>();
    public TreeNode convertBST(TreeNode root) {
         DFS(root);
         for(int i=nodeArray.size()-2;i>=0;i--){
             nodeArray.set(i,nodeArray.get(i+1)+nodeArray.get(i));
         }
         changeNodeVal(root,new LinkedList<>(nodeArray));
         return root;
    }
    private void DFS(TreeNode node){
        if(node==null)
            return;
        DFS(node.left);
        nodeArray.add(node.val);
        DFS(node.right);
    }
    private void changeNodeVal(TreeNode node,Queue<Integer>queue){
        if(node==null)
            return;
        changeNodeVal(node.left,queue);
        node.val= queue.poll();
        changeNodeVal(node.right,queue);
    }
}

题解(二):反序中序遍历,用 int sum 记录当前搜索过的节点的节点值之和,在遍历的过程中就可以进行替换操作

class Solution {
    int sum=0;
    public TreeNode convertBST(TreeNode root) {
         DFS(root);
         return root;
    }
    private void DFS(TreeNode node){
        if(node==null)
            return;
        DFS(node.right);
        sum+=node.val;
        node.val=sum;
        DFS(node.left);
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值