leetcode刷题：二叉树30（把二叉搜索树转换为累加树）

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

力扣题目链接

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

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

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

示例 1：

• 输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
• 输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

示例 2：

• 输入：root = [0,null,1]
• 输出：[1,null,1]

示例 3：

• 输入：root = [1,0,2]
• 输出：[3,3,2]

示例 4：

• 输入：root = [3,2,4,1]
• 输出：[7,9,4,10]

提示：

• 树中的节点数介于 0 和 104 之间。
• 每个节点的值介于 -104 和 104 之间。
• 树中的所有值 互不相同 。
• 给定的树为二叉搜索树。

 1. 把原结点的各值中序遍历存在集合中
 2. 写一个方法，入参为 当前结点的值，输出为sum，内部逻辑是累加数组中当前结点往后的结果
 3. 遍历树，每个结点调用这个方法

package com.programmercarl.tree;

import java.util.ArrayList;
import java.util.List;
import java.util.Stack;

/**
 * @ClassName ConvertBST
 * @Descriotion TODO
 * @Author nitaotao
 * @Date 2022/7/8 9:57
 * @Version 1.0
 * https://leetcode.cn/problems/convert-bst-to-greater-tree/
 * 538. 把二叉搜索树转换为累加树
 **/
public class ConvertBST {
    public TreeNode convertBST(TreeNode root) {
        /**
         * 思路：
         * 1. 把原结点的各值中序遍历存在集合中
         * 2. 写一个方法，入参为 当前结点的值，输出为sum，内部逻辑是累加数组中当前结点往后的结果
         * 3. 遍历树，每个结点调用这个方法
         */
        if (root == null) {
            return root;
        }
        inorderTraversal(root);
        TreeNode node=root;
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(node);
        while (!stack.isEmpty()) {
            node = stack.pop();
            node.val = getSum(node.val);
            if (node.right != null) {
                stack.push(node.right);
            }
            if (node.left != null) {
                stack.push(node.left);
            }
        }
        return root;
    }

    List<Integer> nums = new ArrayList<>();

    /**
     * 中序遍历，保证有序
     *
     * @param root
     */
    public void inorderTraversal(TreeNode root) {
        if (root == null) {
            return;
        }
        inorderTraversal(root.left);
        nums.add(root.val);
        inorderTraversal(root.right);
    }

    public int getSum(int val) {
        int sum = 0;
        for (int i = 0; i < nums.size(); i++) {
            if (nums.get(i) >= val) {
                sum += nums.get(i);
            }
        }
        return sum;
    }
}