问题描述:Given a Binary Search Tree (BST), convert it to a Greater Tree such that every key of the original BST is changed to the original key plus sum of all keys greater than the original key in BST.
实例:Given a binary search Tree `{5,2,3}`:
5
/ \
2 13
Return the root of new tree
18
/ \
20 13
解题思路:给你一颗二叉查找树,把它变成一颗更大的二叉查找树。根据给我们的实例,新的节点等于旧的节点加上原来二叉树比给定节点值大的节点值的和。即如果是位于左子树的节点,新的节点值等于自身节点值加根节点的值加右子树的值。根节点的值等于原根节点的值加右子树的值。左子树的节点值保持不变。具体实现代码可以用递归函数实现。
实现代码:
/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/
class Solution {
public:
/**
* @param root the root of binary tree
* @return the new root
*/
TreeNode* convertBST(TreeNode* root) {
// Write your code here
int sum=0;
convert(root,sum);
return root;
}
void convert(TreeNode *root,int &sum)
{
if (root!=NULL)
{convert(root->right,sum);
root->val+=sum;
sum=root->val;
convert(root->left,sum);
}
}
};
做题感想:进一步体会二叉查找树的特点,当中序遍历二叉查找树的时候节点值正好是从小到大输出的。进一步体会递归函数的使用方法。