leetcode 538. 把二叉搜索树转换为累加树思考分析

题目

给出二叉搜索树的根节点，该树的节点值各不相同，请你将其转换为累加树（Greater Sum Tree），使每个节点 node 的新值等于原树中大于或等于 node.val 的值之和。
提醒一下，二叉搜索树满足下列约束条件：
节点的左子树仅包含键小于节点键的节点。
节点的右子树仅包含键大于节点键的节点。
左右子树也必须是二叉搜索树。
在这里插入图片描述

递归思路

二叉搜索树的中序遍历是一个单调递增的数组，我们要做的就是求出从后到前的累加值：
[2,5,13]->[20,18,13]
数组的从后向前累加转换成二叉搜索树就是反中序遍历，这样就是按照val单调递减的顺序遍历了。
依然需要一个pre指针记录当前遍历结点cur的前一个结点。
递归函数参数以及返回值
遍历整棵树，不需要返回值做操作
定义一个全局变量pre，用来保存cur结点的前一个结点的数值，定义为int型就可以了

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

终止条件
遇到空结点就返回

if(cur == NULL) return ;

单层逻辑
按照右中左来遍历二叉树，中结点的处理逻辑就是让cur的数值加上前一个结点的数值

traversal(cur->right);	//右
cur->val+=pre;
pre=cur->val;
traversal(cur->left);	//左

整体代码：

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     TreeNode *left;*     TreeNode *right;*     TreeNode() : val(0), left(nullptr), right(nullptr) {}*     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}*     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:int pre;	//记录前一个结点的数值void traversal(TreeNode* cur){if(cur == NULL) return ;traversal(cur->right);	//右cur->val+=pre;pre=cur->val;traversal(cur->left);	//左}TreeNode* convertBST(TreeNode* root) {pre = 0;traversal(root);return root;}
};

迭代法

/*** Definition for a binary tree node.* struct TreeNode {*     int val;*     TreeNode *left;*     TreeNode *right;*     TreeNode() : val(0), left(nullptr), right(nullptr) {}*     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}*     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}* };*/
class Solution {
public:int pre;	//记录前一个结点的数值void traversal(TreeNode* root){stack<TreeNode*> st;TreeNode* cur = root;while(cur!=nullptr || !st.empty()){if(cur != nullptr){st.push(cur);cur = cur->right;   //右}else{cur =st.top();st.pop();cur->val+=pre;pre = cur->val;cur = cur->left;}}}TreeNode* convertBST(TreeNode* root) {pre = 0;traversal(root);return root;}
};