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.
Example:
Input: The root of a Binary Search Tree like this: 5 / \ 2 13 Output: The root of a Greater Tree like this: 18 / \ 20 13
题意:
首先需要清楚BST,二叉搜索树的特征:
1. 每个节点都有一个作为搜索依据的关键码( key) , 所有节点的关键码互不相同。
2. 左子树上所有节点的关键码( key) 都小于根节点的关键码( key) 。
3. 右子树上所有节点的关键码( key) 都大于根节点的关键码( key) 。
4. 左右子树都是二叉搜索树。
然后题目要求变化这颗树为更大的树,即每个节点值加上所有大于他的节点值,成为新的值。然后返回这颗变大的树。
解法:由于是二叉搜索树,所以我们知道每棵树上他的所有右节点的值都要比他大,所以采用后序遍历,将右子树的所有节点和求出来然后与自身相加并重新赋值,就能得到更大的树。
class Solution {
public: int Sum = 0;
public:
TreeNode* convertBST(TreeNode* root) {
LrdSum(root);
return root;
}
void LrdSum(TreeNode* root){
if(root == NULL){
return;
}
LrdSum(root->right);
Sum+=root->val;
root->val = Sum;
LrdSum(root->left);
}
};
总的来说知道二叉搜索树(BST)后,这题就没什么难度了。时间复杂度也为O(N);