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

最新推荐文章于 2024-10-09 22:28:58 发布

qq_43701187

最新推荐文章于 2024-10-09 22:28:58 发布

阅读量100

点赞数

文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/qq_43701187/article/details/125600475

版权

解题思路

1、递归法
（1）确定递归终止条件：如果遍历到空结点，则结束返回空；
（2）单层递归逻辑：按右中左的顺序遍历，即对中序遍历稍作改变就可。在中间时加上前一个结点的数值即可。
（3）返回根结点

class Solution {
private:
    int pre; // 记录前一个节点的数值
    void traversal(TreeNode* cur) { // 右中左遍历
        if (cur == NULL) return;
        traversal(cur->right);
        cur->val += pre;
        pre = cur->val;
        traversal(cur->left);
    }
public:
    TreeNode* convertBST(TreeNode* root) {
        pre = 0;
        traversal(root);
        return root;
    }
}

2、迭代法
按右中左的中序遍历的迭代法来做，在处理中间结点时加上前一时刻记录的数值即可

class Solution {
private:
    int pre; // 记录前一个节点的数值
    void traversal(TreeNode* root) {
        stack<TreeNode*> st;
        TreeNode* cur = root;
        while (cur != NULL || !st.empty()) {
            if (cur != NULL) {
                st.push(cur);
                cur = cur->right;   // 右
            } else {
                cur = st.top();     // 中
                st.pop();
                cur->val += pre;
                pre = cur->val;
                cur = cur->left;    // 左
            }
        }
    }
public:
    TreeNode* convertBST(TreeNode* root) {
        pre = 0;
        traversal(root);
        return root;
    }
};