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 node.left != None:
            self.postorder(node.left)
            
    def convertBST(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if root != None:
            self.postorder(root)
        return root