题目:
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(Greater Sum Tree),使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下,二叉搜索树满足下列约束条件:
节点的左子树仅包含键 小于 节点键的节点。
节点的右子树仅包含键 大于 节点键的节点。
左右子树也必须是二叉搜索树。
大致思路:
这道题的题意就是,执行过后,本结点的值,等于把原来所有大于本结点值的结点的值相加起来的和。
由于二叉搜索树的中序遍历是有序且非减的,因为题目写明结点的值唯一,所以可以理解为递增。故中序遍历的反向即为递减。在这一递减序列中,每个数前面的所有数,即为该树中大于本结点值的数。
因此我们可以通过中序遍历后入栈再出栈的形式实现反向遍历,再用一初值为0的变量sum,每出栈一个数就将其与自身相加,再赋给结点,实现结点的累加,代码如下。
注:函数count是用中序遍历计算树中结点的个数,再将其作为栈的大小。但实际执行后对执行效率而言没什么用,可以不用这一个函数,创建栈直接写 10000 大小也是一样的。
void Push_Stake(struct TreeNode*root,struct TreeNode **stack,int *top){//结点入栈函数
if(root==NULL)
return;
Push_Stake(root->left,stack,top);
stack[(*top)++]=root;
Push_Stake(root->right,stack,top);
}
void count_Node(struct TreeNode*root,int* count){//计算树中结点数的函数
if(root==NULL)
return;
count_Node(root->left,count);
(*count)++;
count_Node(root->right,count);
}
struct TreeNode* convertBST(struct TreeNode* root){
struct TreeNode* temp;
int top=0;//栈顶指针
int sum=0;//用于暂存 累加和
int count=0;//用于后面计算树中结点数
count_Node(root,&count);//计算树的结点数
struct TreeNode** stack=malloc(sizeof(struct TreeNode*)*count);//创建栈,栈内元素为结点指针
Push_Stake(root,stack,&top);//用中序遍历入栈
while(top>0){//出栈
temp=stack[--top];
sum+=temp->val;
temp->val=sum;
}
return root;
}