题目
原题链接
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 原始二叉搜索树:
5
/ \
2 13
输出: 转换为累加树:
18
/ \
20 13
思路
看到 二叉搜索树 ,你脑海中最先想到的是什么?
中序遍历!
二叉搜索树的一个非常重要的性质:二叉搜索树的中序遍历是一个递增的有序序列。本题我们需要将其转换为累加树,使得每个节点的值是原来的节点值加上所有大于它的节点值之和。那问题就迎刃而解了,我们只需反向中序遍历即可,并把每次的节点值进行累加,就能得到最终的累加树。
注意,一定是反向中序遍历,也就是 右->中->左 的顺序。
代码,很好理解:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
int sum;
TreeNode* convertBST(TreeNode* root) {
if(!root) return nullptr;
convertBST(root -> right);
root -> val += sum;
sum = root -> val;
convertBST(root -> left);
return root;
}
};