描述
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下,二叉搜索树满足下列约束条件:
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
算法
BST 的特性:
1、对于 BST 的每一个节点node,左子树节点的值都比node的值要小,右子树节点的值都比node的值大。
2、对于 BST 的每一个节点node,它的左侧子树和右侧子树都是 BST。
从大到小遍历,然后用一个全局变量来接
class Solution {
public:
TreeNode* convertBST(TreeNode* root) {
traverse(root);
return root;
}
int sum = 0;
void traverse(TreeNode* node)
{
if (node == nullptr) return;
traverse(node->right);
sum += node->val;
node->val = sum;
traverse(node->left);
}
};