1.题目
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
2.思路
根据题目要求,要把二叉树转化成每一个节点的值是整棵树中大于等于它本身的节点值之和。
由于该树为查找二叉树,因此对于每一个节点:其右子树值转化为本身的值,根节点的值转化为根节点的值加上其右子树的值,左子树的值转化为左子树加上其根节点值加上与它同根的右子树值。
根据以上分析采用递归的算法,最重要的就是sum作为不断改变的求和暂存值,在此叙述其中一步:
递归右子树到最下端的叶子结点,开始相加值,sum暂存其值;
加完右子树回退到根节点,根节点的值+sum赋给sum,根节点转化为sum值;
执行该根节点左子树递归,左子树值+sum赋给sum,左子树的值转化为现sum值。
3.代码
/**
* 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
*/
int sum=0;
TreeNode* convertBST(TreeNode* root) {
// Write your code here
if(root==NULL)
{
return root;
}
convertBST(root->right);
root->val+=sum;
sum=root->val;
convertBST(root->left);
return root;
}
};
4.感想
做这个题的时候深深地怀疑自己不会递归了。感觉绕不出来。。。。
有想法却不知道怎么下手系代码,最后还是忍不住借鉴了网上的做法,搞明白了又自己画了遍图写了一遍。
最重要的就是sum作为不断改变的求和暂存值!