(110)538. 把二叉搜索树转换为累加树(leetcode)-CSDN博客

本文链接：https://blog.csdn.net/li_qw_er/article/details/108702762

题目链接：
https://leetcode-cn.com/problems/convert-bst-to-greater-tree/
难度：简单
538. 把二叉搜索树转换为累加树
给定一个二叉搜索树（Binary Search Tree），把它转换成为累加树（Greater Tree)，使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如：
	输入: 原始二叉搜索树:
	         	 5
 	         	/   \
	  		   2     13
	输出: 转换为累加树:
		         18
		        /   \
    		  20     13

简单题。。。二叉搜索树中序遍历就行了不过要反过来右->中->左

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int flag=0;
    TreeNode* convertBST(TreeNode* root) {
        if(root==nullptr){
            return nullptr;
        }else{
            convertBST(root->right);
            root->val+=flag;
            flag=root->val;
            convertBST(root->left);
        }
        return root;
    }
};

还有种没想到的方法（似乎也没听说过。。。）原理不太懂直接看着文字描述的流程写的（和抄的没啥区别）（感冒好点了但是头晕啊。。。。）

class Solution {
public:
    
    TreeNode* convertBST(TreeNode* root) {
        int sum=0;
        TreeNode* node=root;
        while(node!=nullptr){
            if(node->right==nullptr){
                sum+=node->val;
                node->val=sum;
                node=node->left;
            }else{
                TreeNode* cur=node->right;
                while(cur->left!=nullptr&&cur->left!=node){
                    cur=cur->left;
                }
                if(cur->left==nullptr){
                    cur->left=node;
                    node=node->right;
                }else{
                    cur->left=nullptr;
                    sum+=node->val;
                    node->val=sum;
                    node=node->left;
                }
            }
        }
        return root;
    }
};