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

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

例如:

输入: 二叉搜索树:
              5
            /   \
           2     13

输出: 转换为累加树:
             18
            /   \
          20     13

第一想法是利用中序遍历得到一个递增序列,然后将序列从尾部到首部逐个逐个相加,即得到新的BST的中序遍历序列,再进行遍历修改值即可。

然后发现可以用这样子的方式遍历BST:右孩子-》根节点-》左孩子,即可得出答案。

class Solution {
public:
    TreeNode* convertBST(TreeNode* root) 
    {
        int temp = 0;
        convertBSThelper(root, temp);
        return root;
    }
    void convertBSThelper(TreeNode* root, int& temp)
    {
        if(root == nullptr)
            return;
        convertBSThelper(root->right, temp);
        root->val += temp;
        temp = root->val;
        convertBSThelper(root->left, temp);
    }
};

猜你喜欢

转载自blog.csdn.net/hlk09/article/details/81320714