给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 二叉搜索树: 5 / \ 2 13 输出: 转换为累加树: 18 / \ 20 13
第一想法是利用中序遍历得到一个递增序列,然后将序列从尾部到首部逐个逐个相加,即得到新的BST的中序遍历序列,再进行遍历修改值即可。
然后发现可以用这样子的方式遍历BST:右孩子-》根节点-》左孩子,即可得出答案。
class Solution {
public:
TreeNode* convertBST(TreeNode* root)
{
int temp = 0;
convertBSThelper(root, temp);
return root;
}
void convertBSThelper(TreeNode* root, int& temp)
{
if(root == nullptr)
return;
convertBSThelper(root->right, temp);
root->val += temp;
temp = root->val;
convertBSThelper(root->left, temp);
}
};