Leetcode 1038. 把二叉搜索树转换为累加树

题目描述

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/binary-search-tree-to-greater-sum-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

C++

/**
 * 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:
     TreeNode* pre; //当作游标，遍历时的前一个节点。
    TreeNode* bstToGst(TreeNode* root) {
        if(!root) return NULL;
        root->right=bstToGst(root->right);
        if(pre) root->val=root->val+pre->val;
        pre=root;
        root->left=bstToGst(root->left);
        return root;
    }
  
};