代码随想录二叉树——把二叉搜索树转换为累加树

题目

给出二叉 搜索 树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node的新值等于原树中大于或等于 node.val 的值之和。

提醒一下，二叉搜索树满足下列约束条件：

节点的左子树仅包含键 小于 节点键的节点。 节点的右子树仅包含键 大于 节点键的节点。 左右子树也必须是二叉搜索树。

示例 1：

输入：[4,1,6,0,2,5,7,null,null,null,3,null,null,null,8]
输出：[30,36,21,36,35,26,15,null,null,null,33,null,null,null,8]

示例 2：

输入：root = [0,null,1] 输出：[1,null,1]

示例 3：

输入：root = [1,0,2] 输出：[3,3,2]

示例 4：

输入：root = [3,2,4,1] 输出：[7,9,4,10]

思路

累加树：当前节点的值，等于比自己大的节点值的总和加上自身的节点值，类似于有序数组求累加，数组位置i的累加和即为从位置i（包括位置i）一直到结尾的所有值之和

二叉搜索树是有序的，而对于有序数组求累加，直接从后往前遍历，挨个累加即可，如[2,5,13]，求从后到前的累加数组就是[20,18,13]

而对于二叉树的话，树中求累加的顺序是右中左，所以只需要反向遍历这个二叉树，顺序累加就可以了

1、递归法：

需要一个pre记录当前遍历节点cur的前一个节点（这里的前一个节点指的是累加的顺序，即树中右边比自己大的相邻的节点），方便做累加

（1）递归函数参数和返回值：不需要返回值，因为要遍历整棵树

int pre;// 记录前一个节点的数值
void traversal(TreeNode cur)

（2）递归终止条件：遇到空就终止

if(cur == null) return;

（3）确定单层递归逻辑：用右中左来遍历二叉树，中节点的处理逻辑就是让cur的数值加上前一个节点pre的数值

traversal();//右

cur.val += pre;//中
pre = cur.val;//当前节点加上后重新赋值给pre，继续后续累加

traversal();//左

完整代码如下：

class Solution {
	int pre;//记录前一个节点(右边比自己大的节点)的值
	public TreeNode convertBST(TreeNode root){
		pre = 0;
		convertBST1(root);
		return root;
	}

	//按右中左顺序遍历，累加即可
	public void convertBST1(TreeNode root){
		if(root == null) return;//递归终止条件
		//单层递归逻辑
		convertBST1(root.right);//右
		pre += root.val;//当前节点加上前面节点(右边比自己大的节点)的值并把累加值存在pre中
		root.val = pre;
		convertBST1(root.left);//左
	}
}

2、迭代法：

迭代法相当于中序模板题