leetcode 538. 把二叉搜索树转换为累加树思考分析

最新推荐文章于 2023-01-08 16:48:27 发布

拾牙慧者

最新推荐文章于 2023-01-08 16:48:27 发布

阅读量162

点赞数 1

分类专栏： # 算法：二叉树基础 leetcode刷题笔记文章标签：二叉树算法 leetcode 数据结构

本文链接：https://blog.csdn.net/qq_42604176/article/details/109400780

版权

leetcode刷题笔记同时被 2 个专栏收录

83 篇文章 3 订阅

订阅专栏

算法：二叉树基础

23 篇文章 0 订阅

订阅专栏

这篇博客讨论了如何将一个二叉搜索树转换为累加树，使得每个节点的值等于原树中大于或等于其值的所有节点的值之和。提供了两种解法：递归和迭代。递归方法通过反中序遍历实现，而迭代方法则使用栈辅助完成。两种方法都遵循右中左的顺序更新节点值。

摘要由CSDN通过智能技术生成

题目

给出二叉搜索树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下，二叉搜索树满足下列约束条件：
节点的左子树仅包含键小于节点键的节点。
节点的右子树仅包含键大于节点键的节点。
左右子树也必须是二叉搜索树。
在这里插入图片描述

递归思路

二叉搜索树的中序遍历是一个单调递增的数组，我们要做的就是求出从后到前的累加值：
[2,5,13]->[20,18,13]
数组的从后向前累加转换成二叉搜索树就是反中序遍历，这样就是按照val单调递减的顺序遍历了。
依然需要一个pre指针记录当前遍历结点cur的前一个结点。
递归函数参数以及返回值
遍历整棵树，不需要返回值做操作
定义一个全局变量pre，用来保存cur结点的前一个结点的数值，定义为int型就可以了

int pre;	//记录前一个结点的数值
void traversal(TreeNode* cur)

终止条件
遇到空结点就返回

if(cur == NULL) return ;

单层逻辑
按照右中左来遍历二叉树，中结点的处理逻辑就是让cur的数值加上前一个结点的数值

traversal(cur->right);	//右
cur->val+=pre;
pre=cur->val;
traversal(cur->left);	//左

整体代码：

/**
 * 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 pre;	//记录前一个结点的数值
    void traversal(TreeNode* cur)
    {
        if(cur == NULL) return ;
        traversal(cur->right);	//右
        cur->val+=pre;
        pre=cur->val;
        traversal(cur->left);	//左
    }
    TreeNode* convertBST(TreeNode* root) {
        pre = 0;
        traversal(root);
        return root;
    }
};

迭代法

/**
 * 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 pre;	//记录前一个结点的数值
    void traversal(TreeNode* root)
    {
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while(cur!=nullptr || !st.empty())
        {
            if(cur != nullptr)
            {
                st.push(cur);
                cur = cur->right;   //右
            }
            else
            {
                cur =st.top();
                st.pop();
                cur->val+=pre;
                pre = cur->val;
                cur = cur->left;
            }
        }
    }
    TreeNode* convertBST(TreeNode* root) {
        pre = 0;
        traversal(root);
        return root;
    }
};