1038. 把二叉搜索树转换为累加树

luckycoding

于 2021-07-14 00:02:25 发布

阅读量82

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/luckycoding/article/details/118714025

leetcode 专栏收录该内容

21 篇文章

订阅专栏

该博客讨论了一种将二叉搜索树（BST）转换为全局最大堆（GST）的方法。通过递归地处理每个节点，使得每个节点的值等于其本身加上其右子树的累加和中的最大值。算法首先检查右子树，然后处理左子树，以构建新的GST树。

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

当前节点下

1.如果有右子树，则先递归右子树。得出右子树累加和。当前节点为和加最大值。

2.如果无右子树，则当前值为原树值+MAX（左子树和，父节点值）。

3.如果左子树，则递归左子树。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int solve(TreeNode *root, TreeNode **ans, int sum) {
        if (root == NULL)
            return 0;
        //cout << root->val << endl;
        int tmp = 0;
        *ans = new TreeNode(0);
        if (root->right != NULL)
            tmp = solve(root->right, &((*ans)->right), sum);
        
        (*ans)->val = root->val + max(tmp, sum);
        tmp = (*ans)->val;
        if (root->left != NULL)
            tmp = solve(root->left, &((*ans)->left), tmp);
        return tmp;
    }
    TreeNode* bstToGst(TreeNode* root) {
        TreeNode *ans;
        solve(root, &ans, 0);
        return ans;
    }
};