给定BST。改动BST,使得每一个点都是大于他的结点的值之和
关键是这题递归參数怎么设计,每一个点比他大的有两快。一个是右子书(假设有的话),还有一个是祖先里面比他大的,假设直接用这两个的话,找不到递归结构。
应该设计第一个參数是祖先比他大的节点和,第二个參数是子树的和。包含自身。
struct Node
{
int val;
Node* left, *right;
};
class Solution
{
void GreaterSumBST(Node* root){
int SubTreeSum=0;
F(root, 0, SubTreeSum);
}
void F(Node* root, int AncestorGreaterSum, int& SubTreeSum){
if(!root){
SubTreeSum=0;
return ;
}
int LeftSubTreeSum=0, RightSubTreeSum=0;
F(root->right, AncestorGreaterSum, RightSubTreeSum);
F(root->left, AncestorGreaterSum+root->val+RightSubTreeSum, LeftSubTreeSum);
SubTreeSum=LeftSubTreeSum+RightSubTreeSum;
root->val=AncestorGreaterSum+RightSubTreeSum;
}
};
时间复杂度ON,空间O1,此处不包括系统栈