538. 把二叉搜索树转换为累加树
作者:互联网
给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。
例如:
输入: 二叉搜索树: 5 / \ 2 13 输出: 转换为累加树: 18 / \ 20 13
思路:
用反序的中序遍历遍历每个结点,同时记录已遍历的所有结点的和sum
,遍历到某个结点时将sum
加到val
。
题解:
/**
* 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 sum = 0;
TreeNode* convertBST(TreeNode* root) {
if (root==NULL) {
return NULL;
}
convertBST(root->right);
root->val += sum;
sum = root->val;
convertBST(root->left);
return root;
}
};
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/convert-bst-to-greater-tree
标签:NULL,TreeNode,val,sum,二叉,遍历,累加,root,538 来源: https://blog.csdn.net/acticn/article/details/104087933