Leetcode第538题把二叉搜索树转换为累加树|C语言

题目:
给出二叉 搜索 树的根节点,该树的节点值各不相同,请你将其转换为累加树(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;
}

执行情况

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值