给定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,此处不包含系统栈