算法力扣刷题记录六十二【538.把二叉搜索树转换为累加树】

danaaaa

于 2024-07-28 09:34:22 发布

阅读量341

点赞数 7

分类专栏：算法小白力扣刷题记录文章标签：算法 leetcode 职场和发展 c++

本文链接：https://blog.csdn.net/danaaaa/article/details/140746052

版权

算法小白力扣刷题记录专栏收录该内容

68 篇文章 2 订阅

订阅专栏

前言

终于二叉树篇最后一道题。可以总结一个章节了。
记录六十二【538.把二叉搜索树转换为累加树】

一、题目阅读

给出二叉搜索树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键小于节点键的节点。
节点的右子树仅包含键大于节点键的节点。
左右子树也必须是二叉搜索树。

示例 1：
在这里插入图片描述

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

示例 2：
在这里插入图片描述

输入：root = [0,null,1]
输出：[1,null,1]

示例 3：
在这里插入图片描述

输入：root = [1,0,2]
输出：[3,3,2]

示例 4：
在这里插入图片描述

输入：root = [3,2,4,1]
输出：[7,9,4,10]

提示：

树中的节点数介于 0 和 10^4 之间。
每个节点的值介于 -10^4 和 10^4 之间。
树中的所有值 互不相同 。
给定的树为二叉搜索树。

二、尝试实现

思路

理解题目：树的结构没有改变；改变的是节点的值。把节点的值换成——原树中大于等于该节点原有值之和。
二叉搜索树中，比当前节点值大的节点在它自身的右子树中，或者是它从属子树的当前节点及右子树。那么一个统一的规律：如果先深入右子树遍历，向上一层返回修改之后的值，并且记录下此时的求和。到上一层后，root->val += sum。如此循环，可以修改整个树的值。（如图所示）
上面统一的规律就是递归的逻辑。开始递归函数实现：
递归参数：TreeNode*当前节点。
递归返回值：TreeNode* 修改之后的子树，返回上一层；
递归终止条件：遇到空节点，return nullptr。无需修改；
递归逻辑：题干说大于等于原有值求和，所以先进右子树递归。

先递归右子树；返回修改之后的子树节点；
修改当前节点：root->val += sum。sum = root->val。更新sum
递归左子树。
return root。

代码实现【递归+sum变量】

class Solution {
public:
    int sum = 0;
    TreeNode* convertBST(TreeNode* root) {
        //终止条件
        if(!root) return nullptr;

        root->right = convertBST(root->right);
        //回到当前层，处理中间节点
        root->val += sum;
        sum = root->val;
        root->left = convertBST(root->left);
        return root;
    }
};

三、参考学习

参考学习链接

学习内容

思路肯定是一致的。但是从有序数组中想也是一个方法。但是二、中代码实现用int sum记录求和，可是参考给出pre、cur双指针的操作。
代码实现【递归+双指针】：总之——本质都是一样的：int或者TreeNode* 都可以，返回值类型看习惯，新定义函数调用或者借助主函数。

class Solution {
public:
    TreeNode* pre = nullptr;
    TreeNode* convertBST(TreeNode* root) {
        //终止条件
        if(!root) return nullptr;

        root->right = convertBST(root->right);
        //回到当前层，处理中间节点
        if(pre){
            root->val += pre->val;
        }
        pre = root;
        root->left = convertBST(root->left);
        return root;
    }
};

迭代法：遍历顺序是右中左。使用迭代模版实现，个人习惯使用记录三十六【二叉树迭代遍历】中统一前中后序的迭代模版，就是空指针标记的方法：

class Solution {
public:
    TreeNode* convertBST(TreeNode* root) {
        stack<TreeNode*> st;
        int sum = 0;
        if(!root) return nullptr;
        st.push(root);
        while(!st.empty()){
            TreeNode* cur = st.top();st.pop();
            if(cur){
                if(cur->left) st.push(cur->left);
                st.push(cur);
                st.push(nullptr);
                if(cur->right) st.push(cur->right);
            }else{
                TreeNode* node = st.top();st.pop();
                node->val += sum;
                sum = node->val;
            }
        }
        return root;
    }
};