Convert BST to Greater Tree

最新推荐文章于 2024-09-03 15:28:22 发布

原创最新推荐文章于 2024-09-03 15:28:22 发布 · 672 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种将二叉搜索树转换为更大的二叉树的方法，使得每个节点的值变为原始值加上所有大于该值的节点之和。通过一次遍历实现，保持了树的性质。

摘要生成于 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

题目意思：

给一个二叉搜索树，将它转化为更大的二叉树，使得节点的值为原始节点的值加上所有比它大的节点的值之和，并且使转化后的二叉树的左子树大于右子树。

解题思路：

设置一个数组，对二叉树进行中序遍历，用数组保存这些值，得到从小到大的一个数组；再次对二叉树进行中序遍历，将所遍历节点的值加上所有比它大的节点的值；对数组进行反转，即让它的值从大到小排序，然后对每个节点求和后将数组最后一个元素丢掉，因为最大的那个值不用变。

也可以采用一次 bst 遍历，然后进行递归调用。

代码如下：

class Solution {
public:
int s;
void work(TreeNode* root) {
if(!root) return;
if(!root->left && !root->right) {
root->val += s;
s = root->val;
return;
}
if(root->right) work(root->right);
root->val += s;
s = root->val;
work(root->left);
}
TreeNode* convertBST(TreeNode* root) {
s = 0;
work(root);
return root;
}
};