Convert BST to Greater Tree

obudunkey

于 2017-04-27 19:50:13 发布

阅读量339

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/DDVAL/article/details/70877374

leetcode 专栏收录该内容

18 篇文章

订阅专栏

本文介绍了一种算法，用于将二叉搜索树（BST）转换为Greater Tree，转换后的树每个节点的值等于原节点值加上所有大于该节点值的节点之和。通过中序遍历实现，最终返回转换后的树。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述：

Given a Binary Search Tree (BST), convert it to a Greater Tree such that every key of the original BST is changed to the original key plus sum of all keys greater than the original key in BST.

Example:

Input: The root of a Binary Search Tree like this:
              5
            /   \
           2     13

Output: The root of a Greater Tree like this:
             18
            /   \
          20     13

思路：

中序遍历

代码：

class Solution {
public:
TreeNode* convertBST(TreeNode* root) {
if (root==NULL)
return NULL;
vector<TreeNode*> res;
inorder(root,res);
int n=res.size();
for (int i=n-2; i>=0; i--)
(res[i])->val=(res[i])->val+(res[i+1])->val;
return root;
}

void inorder(TreeNode* node, vector<TreeNode*>& res)
{
if (node==NULL)
return;
inorder(node->left,res);
res.push_back(node);
inorder(node->right,res);
}
};