LeetCode-538 Convert BST to Greater Tree

题目链接

题目大意为：

给一个二叉搜索树（BST），生成新树，树的每个值都要加上树中所有比它大的值。

5
/   \
2     13

比5大的有13，比2 大的有13和5，所以生成的新树为

18（5+13）
/   \
20     13

第一次写的时候居然没想起二叉搜索树的概念，，，基本功忘得太快了，，然后当成普通二叉树做的，将所有节点加入vector中，然后从大到小排序，从第2项开始，依次加上前一项，最后返回root，假如卡内存估计就GG了，代码如下：

class Solution {
public:
vector<TreeNode*> dic;
void dfs(TreeNode* node) {
dic.push_back(node);
if (node->left != NULL) {
dfs(node->left);
}
if (node->right != NULL) {
dfs(node->right);
}
}
TreeNode* convertBST(TreeNode* root) {
dic.clear();
if (root != NULL)
dfs(root); // 将所有节点加入容器中
for (int i = 0; i < (int)dic.size(); i++) { // 排序
for (int j = 0; j < i; j++) {
if (dic[i]->val > dic[j]->val) {
swap(dic[i], dic[j]);
}
}
}
for (int i = 1; i < (int)dic.size(); i++)   // 依次加前项，相当于累加和{
dic[i]->val += dic[i - 1]->val;
}
return root;
}
};

ac后感觉怪怪的，自己绝对又犯傻了，，，查了下百度百科二叉搜索树概念，右子节点大于父节点大于左子节点。。。这么基础的我居然忘了，，mmp，然后改了改上述代码，既然右节点一定大于父节点，那就可以先走右，再加到父节点上，再加到左子节点上，这顺序不就是后序遍历吗，所以，，，v2.0就出来了。。。

class Solution(object):
sum = 0
def postorder(self, node):
if node.right != None:
self.postorder(node.right)

node.val += self.sum
self.sum = node.val

if
4000
node.left != None:
self.postorder(node.left)

def convertBST(self, root):
"""
:type root: TreeNode
:rtype: TreeNode
"""
if root != None:
self.postorder(root)
return root