Leetcode 538 - Convert BST to Greater Tree

最新推荐文章于 2023-02-25 03:31:14 发布

原创最新推荐文章于 2023-02-25 03:31:14 发布 · 394 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 同时被 2 个专栏收录

157 篇文章

订阅专栏

tree

16 篇文章

订阅专栏

本文介绍了一种特殊的二叉搜索树(BST)转换方法，通过改变节点值使其等于该节点及其右侧所有节点值之和。利用BST的性质进行逆中序遍历，实现了简洁高效的算法。

题意

给一个BST，改变其节点的值，将其节点的值加上所有比他大的节点的值

思路

首先注意是BST，其中序遍历的节点是递增序列。即假设中序遍历后的结果是 $a_0, a_1, a_2, .... a_n$ ，我们有 $a_0 < a_1 < a_2 < ... < a_n$ 。

那么对于 $a_i$ ，其改变后的值应该是 $a_i + a_{i+1} + a_{i + 2} + ... + a_n$ ，即我们只需要将 $a_i$ 之后遍历的节点值累加起来即可。为了达到这个效果，我们只需要将中序遍历的顺序倒过来： $a_n, a_{n - 1}, ... , a_2, a_1, a_0$ ，对于 $a_i$ ，只需要累加上其之前遍历过的节点值即可。

所以，我们只需要右，根，左的去遍历节点，并且遍历到根节点时，累加上之前遍历过的节点值即可。

代码

/**
 * 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;
    void dfs(TreeNode* root) {
        if (!root) return;
        dfs(root->right);
        sum += root->val;
        root->val = sum;
        dfs(root->left);
    }

    TreeNode* convertBST(TreeNode* root) {
        dfs(root);
        return root;
    }
};