• 首页

  • 文章归档

  • 文章分类

  • 日志

  • 图库

  • 友链

  • 留言板

  • 关于我
H i , m e g u m i
H i , m e g u m i

无名高

获取中...

09
21
算法题

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

发表于 2020-09-21 • 树 leetcode Morris遍历 • 被 403 人看爆

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

 

image.png

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-bst-to-greater-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。


BST的中序遍历结果是从小到大的排列,反序的中序遍历就是从大到小的排列。
我们使用反序的中序遍历,在遍历到某个节点时,可以得到大于当前节点的所有值,加上该值即可。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
private:
    int sum = 0;
public:
    /**\
     给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
     **/
    TreeNode* convertBST(TreeNode* root) {
        if (root != nullptr) {
            convertBST(root->right);
            root->val += sum;
            sum = root->val;
            convertBST(root->left);
        }
        return root;
    }
    
    
};

同样这道题可以使用Morris遍历
关于Morris遍历可以参考我的另一篇笔记 https://www.wuminggao.cn/archives/leetcode94
对于反序的中序遍历,那么一个节点(有右子树的情况下)的前驱节点就是右子树的最左节点

代码:

class Solution {
private:
    int sum = 0;
public:
    /**\
     给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
     **/
    TreeNode* convertBST(TreeNode* root) {
        int sum = 0;
        TreeNode* tmp = root;
        TreeNode* predecessor = nullptr;
        while (tmp != nullptr) {
            if (tmp->right == nullptr) {
                sum += tmp->val;
                tmp->val = sum;
                tmp = tmp->left;
            }else{
                // 存在右子树 找到右子树的最左节点
                predecessor = tmp->right;
                while (predecessor->left != nullptr && predecessor->left != tmp) {
                    predecessor = predecessor->left;
                }
                // 右子树已经访问完毕,处理当前节点,然后转向左子树
                if (predecessor->left != nullptr) {
                    sum += tmp->val;
                    tmp->val = sum;
                    predecessor->left = nullptr;
                    tmp = tmp->left;
                }else{
                    predecessor->left = tmp;
                    tmp = tmp->right;
                }
            }
        }
        return root;
    }

    
};
分享到:
LeetCode 617. 合并二叉树
LeetCode 78. 子集
  • 文章目录
  • 站点概览
无名高

帅哥无名高

我们是如何走到这一步

Github QQ Email RSS
看爆 Top5
  • SpringBoot学习笔记 2,395次看爆
  • 解决在IDEA编写Java代码时,向数据库中插入中文字符后显示?乱码问题 2,324次看爆
  • JDBC核心技术笔记 2,204次看爆
  • SpringMVC学习笔记 1,907次看爆
  • MySQL基础学习笔记 1,860次看爆
粤公网安备 44030702003128号 黔ICP备20006240号

Copyright © 2022 无名高 ·

Proudly published with Halo · Theme by fyang · 站点地图