一、题目描述
给定一个二叉搜索树,请将它的每个节点的值替换成树中大于或者等于该节点值的所有节点值之和。
输入:root = [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]
二、代码思路
BVL: 二叉搜索树,其特点就是左大于根,根大于左,如此递归。
所以,我们找大于某个节点的所有节点,就找该节点的左子树即可。
2. 如果该节点是左子树左节点,找左子树即可。
4. 如果是左子树右节点,找左子树以及他爹的左子树。
5. 如果是右子树左节点,找左以及根节点所有左节点。
6. 如果是右子树右节点,找左以及前面遍历过的所有节点,因为其实他是最小的。
我们不难发现,其实按照根据BVL的特点,我们中序遍历,就是从大到小遍历的二叉树。
所以,我们只需要中序遍历即可,并记录每个比目标值大的值即可。但是要注意等于的情况。
三、 代码题解
class Solution {
private int sum = 0;
private void dfs(TreeNode node) {
if (node == null) return;
dfs(node.right);
int temp = node.val;
node.val += sum;
sum += temp;
dfs(node.left);
}
public TreeNode convertBST(TreeNode root) {
//BVL: 二叉搜索树,其特点就是左大于根,根大于左,如此递归。
//所以,我们找大于某个节点的所有节点,就找该节点的左子树即可。
//2. 如果该节点是左子树左节点,找左子树即可。
//3. 如果是左子树右节点,找左子树以及他爹的左子树。
//4. 如果是右子树左节点,找左以及根节点所有左节点。
//5. 如果是右子树右节点,找左以及前面遍历过的所有节点,因为其实他是最小的。
//我们不难发现,其实按照根据BVL的特点,我们中序遍历,就是从大到小遍历的二叉树。
//所以,我们只需要中序遍历即可,并记录每个比目标值大的值即可。但是要注意等于的情况。
dfs(root);
return root;
}
}