Leetcode538. 把二叉搜索树转换为累加树(逆中序遍历)

while_false_

于 2019-05-17 15:54:22 发布

阅读量167

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode题解

本文链接：https://blog.youkuaiyun.com/qq_34595352/article/details/90294773

LeetCode题解专栏收录该内容

101 篇文章

订阅专栏

博客给出链接并介绍将二叉搜索树转换为累加树的问题，即让每个节点的值变为原来节点值加上所有大于它的节点值之和，还给出了输入输出示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

https://leetcode-cn.com/problems/convert-bst-to-greater-tree/

给定一个二叉搜索树（Binary Search Tree），把它转换成为累加树（Greater Tree)，使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

例如：

输入: 二叉搜索树:
5
/ \
2 13

输出: 转换为累加树:
18
/ \
20 13

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

/*思路
逆中序，则将数排列为才从大到小的一个数组，计算累加值即可
注意下sum 加上的原root->val,不是变化后的root->val
*/

void reverse_middle_sort(struct TreeNode* root, int *sum){
    if(!root) return;
    reverse_middle_sort(root->right, sum);
    int tmp = root->val;
    root->val += *sum;
    *sum += tmp;
    reverse_middle_sort(root->left, sum);
}

struct TreeNode* convertBST(struct TreeNode* root){
    int sum = 0;
    reverse_middle_sort(root, &sum);
    return root;
}