538.把二叉搜索树转换为累加树
一、题目描述
给定一个二叉搜索树 root (BST),请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。
提醒一下, 二叉搜索树 满足下列约束条件:
- 节点的左子树仅包含键 小于 节点键的节点。
- 节点的右子树仅包含键 大于 节点键的节点。
- 左右子树也必须是二叉搜索树。
示例 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]
提示:
- 树中的节点数在 [1, 100] 范围内。
- 0 <= Node.val <= 100
- 树中的所有值均 不重复 。
二、解题思路
从树中可以看出累加的顺序是右中左,所以我们需要反中序遍历这个二叉树,然后顺序累加就可以了。
三、AC代码
递归法
class Solution {
int tmp = 0;
public TreeNode convertBST(TreeNode root) {
if (root == null) {
return null;
}
TreeNode right = convertBST(root.right);
tmp += root.val;
root.val = tmp;
TreeNode left = convertBST(root.left);
return root;
}
}
参考代码随想录:https://programmercarl.com/