2022-8-17 剑指offer-二叉树-递归

剑指 Offer II 054. 所有大于等于节点的值之和

难度中等35收藏分享切换为英文接收动态反馈

给定一个二叉搜索树，请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。

 

提醒一下，二叉搜索树满足下列约束条件：

• 节点的左子树仅包含键 小于 节点键的节点。
• 节点的右子树仅包含键 大于 节点键的节点。
• 左右子树也必须是二叉搜索树。

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    List<TreeNode> list;
    public TreeNode convertBST(TreeNode root) {
        list=new ArrayList<>();
        inorder(root);
        int sum=0;
        for (int i=list.size()-1;i>=0;i--){
            TreeNode temp=list.get(i);
            sum+=temp.val;
            temp.val=sum;
        }
        return root;
    }

    public void inorder(TreeNode root){
        if (root==null) return;
        inorder(root.left);
        list.add(root);
        inorder(root.right);
    }
}

思路：反序中序遍历，累加值更新即可。