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

╰半橙微兮° 2023-02-14 03:09 3阅读 0赞

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

*  1、问题分析
 *  2、问题解决
 *  3、总结

# 1、问题分析 #

题目链接：[https://leetcode-cn.com/problems/convert-bst-to-greater-tree/][https_leetcode-cn.com_problems_convert-bst-to-greater-tree]  
  本质上就是一个二叉树的遍历问题。代码我已经进行了详细的注释，理解应该没有问题，读者可以作为参考，如果看不懂(可以多看几遍)，欢迎留言哦！我看到会解答一下。

# 2、问题解决 #

笔者以`C++`方式解决。

#include "iostream"
    
    using namespace std;
    
    #include "algorithm"
    #include "vector"
    #include "queue"
    #include "set"
    #include "map"
    #include "string"
    #include "stack"
    
    /** * Definition for a binary tree node. * struct TreeNode { * int val; * TreeNode *left; * TreeNode *right; * TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */
    
    struct TreeNode { 
        int val;
        TreeNode *left;
        TreeNode *right;
    
        TreeNode(int x) : val(x), left(NULL), right(NULL) { }
    };
    
    class Solution { 
    private:
        // 将二叉树变成数组
        // 存储中序遍历节点的值
        // 中序遍历对于二叉搜索树正好是递增序列
        vector<int> dp;
    public:
        TreeNode *convertBST(TreeNode *root) { 
            inorder(root);
            for (int i = dp.size() - 2; i >= 0; --i) { 
                dp[i] = dp[i] + dp[i + 1];
            }
            int index = 0;
            inorder2(root, index);
            return root;
        }
    
        // 根据中序顺序将节点存储到数组中
        void inorder(TreeNode *root) { 
            if (root != NULL) { 
                inorder(root->left);
                dp.push_back(root->val);
                inorder(root->right);
            }
        }
    
        /** * 中序遍历，将 dp 数组中的值，赋值到二叉搜索树中 * @param root * @param index 这里是引用类型，因为需要修改 index 的值，不断递归，递归结束后 index * 就不是传递进去的值 */
        void inorder2(TreeNode *root, int &index) { 
            if (root != NULL) { 
                inorder2(root->left, index);
                // 中序赋值
                root->val = dp[index];
                // 赋完值后，需要将数组下标 + 1
                index += 1;
                inorder2(root->right, index);
            }
        }
    };
    
    void printInorder(TreeNode *root) { 
        if (root != NULL) { 
            printInorder(root->left);
            cout << root->val << " ";
            printInorder(root->right);
        }
    }
    
    int main() { 
    
    // TreeNode *pNode2 = new TreeNode(2);
    // TreeNode *pNode5 = new TreeNode(5);
    // TreeNode *pNode13 = new TreeNode(13);
    // TreeNode *pNode7 = new TreeNode(7);
    // pNode5->left = pNode2;
    // pNode5->right = pNode13;
    // pNode13->left = pNode7;
    
        TreeNode *pNode1 = new TreeNode(2);
        TreeNode *pNode2 = new TreeNode(0);
        TreeNode *pNode3 = new TreeNode(3);
        TreeNode *pNode4 = new TreeNode(-4);
        TreeNode *pNode5 = new TreeNode(1);
    
        pNode1->left = pNode2;
        pNode1->right = pNode3;
    
        pNode2->left = pNode4;
        pNode2->right = pNode5;
    
    
        Solution *pSolution = new Solution;
        TreeNode *pNode = pSolution->convertBST(pNode1);
        printInorder(pNode);
        cout << endl;
    
    
        system("pause");
        return 0;
    }

运行结果

![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3_size_16_color_FFFFFF_t_70]  
有点菜，有时间再优化一下。

# 3、总结 #

难得有时间刷一波`LeetCode`, 这次做一个系统的记录，等以后复习的时候可以有章可循，同时也期待各位读者给出的建议。算法真的是一个照妖镜，原来感觉自己也还行吧，但是算法分分钟教你做人。前人栽树，后人乘凉。在学习算法的过程中，看了前辈的成果，受益匪浅。  
感谢各位前辈的辛勤付出，让我们少走了很多的弯路！  
哪怕只有一个人从我的博客受益，我也知足了。  
点个赞再走呗！欢迎留言哦！

[https_leetcode-cn.com_problems_convert-bst-to-greater-tree]: https://leetcode-cn.com/problems/convert-bst-to-greater-tree/
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/20200531222029645.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3FxXzMyNTEwNTk3,size_16,color_FFFFFF,t_70