LeetCode538. Convert BST to Greater Tree

Convert BST to Greater Tree

1、原题

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.

2、题目解析

题目原意是说，给我们一个二叉查找树，然后要求我们对二叉查找树上的每一个数字进行修改，其值为原值加上树种比其大的数。

     思考：第一眼看到这题，第一个想法就是对每一个节点遍历一边树，找到比其大的数字。但是感觉有些繁琐了些。于是又仔细想了想题目。发现题目中明确提到了是二叉查找树，这样的数有一个特点，就是其节点左边的都比该节点的值要小，节点右边的都比该节点的值要大。于是我们可以根据其特征，修改我们的算法。算法思路如下：

     1、初始化sum值为0；

     2、访问节点的右节点，直到其右节点不存在。

     3、设置该右节点的值current.val += sum；设置sum值，sum = current.val(当前节点值);    

     4、访问左节点，重复刚才过程。

    具体原理就是因为二叉查找树种比一节点数要大的数都在其右边，我们通过递归的方法获取其所有右节点的累加和即可。

3、代码实现

public class Solution {
    //初始化
	private int sum = 0;
	
	//递归寻找其大值之和
	public void post(TreeNode current) {
		if (current != null) {
			post(current.right);
			current.val += sum;
			sum = current.val;
			post(current.left);
		}
	}
	
	public TreeNode convertBST(TreeNode root) {
		post(root);
        return root;
    }
}