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

1、问题分析

题目链接：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;
}

运行结果

有点菜，有时间再优化一下。

3、总结

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