Leetcode538. 采用中序遍历将BST转化为节点值更大的树

Leetcode538. Convert BST to Greater Tree

题目

Given a Binary Search Tree (BST), convert it to a Greater Tree such that every key of the original BST is changed to the original key plus sum of all keys greater than the original key in BST.

Example:
Input: The root of a Binary Search Tree like this:
5
/ \
2 13
Output: The root of a Greater Tree like this:
18
/ \
20 13

解题分析

题目乍一看，很可能被懵住了，在想怎么遍历才能使得左子树的节点的值能够加上右边节点的和。但我们不妨想一下，是否可以将树转化为一个数组，并将其排序？这样我们就可以对数组进行遍历然后依次转化它们的值，最后只需再按顺序将数组元素的值插入到树中就可以了。

我们知道，二叉搜索树有这样一个性质：某个节点的左子树的所有节点的值都会小于该节点，而右子树的所有节点的值都会大于该节点；而中序遍历刚好就是先遍历左子树，再遍历中间节点，最后再遍历右子树。

想到了这一点，问题就解决一大半了。注意中间有一点，我们知道，vector是没有pop_front()的函数的，所以要想删除第一个元素，就必须用迭代器，这显然有点麻烦了；在这里我们做一个小小的变换，将数组颠倒过来，最后再中序遍历往树中插值的时候就可以取出最后一个值并将其pop_back()，这样就省去了一些不必要的麻烦。

源代码

/** * Definition for a binary tree node. * struct TreeNode { *     int val; *     TreeNode *left; *     TreeNode *right; *     TreeNode(int x) : val(x), left(NULL), right(NULL) {} * }; */class Solution {public:    TreeNode* convertBST(TreeNode* root) {        if(root == NULL) {            return NULL;        }        int i, j;        vector<int> v;        inorder(root, v);        for(i = 0, j = v.size() - 1; i < j; j--) {            v[i] += v[j];            if(i == j - 1) {                i++;                j = v.size();            }        }        reverse(v.begin(), v.end());        convert(root, v);        return root;    }    void inorder(TreeNode* root, vector<int>& v) {        if(root == NULL) {            return;        }        inorder(root->left, v);        v.push_back(root->val);        inorder(root->right, v);    }    void convert(TreeNode* root, vector<int>& v) {        if(root == NULL) {            return;        }        convert(root->left, v);        root->val = v.back();        v.pop_back();        convert(root->right, v);    }};

这只是我对这道题的一些想法，有问题还请在评论区讨论留言~