所有大于等于当前节点的值java实现

最新推荐文章于 2024-10-06 20:16:20 发布

鬼谷春秋

最新推荐文章于 2024-10-06 20:16:20 发布

阅读量182

点赞数

分类专栏： java 文章标签： java 哈希算法开发语言

本文链接：https://blog.csdn.net/guiguchunqiu/article/details/121462649

版权

java 专栏收录该内容

107 篇文章 1 订阅

订阅专栏

该博客介绍了一种算法，用于将给定的二叉搜索树（BST）转换为累加树，其中每个节点的值等于其原始值加上所有大于等于它的节点的值。算法使用栈来控制遍历顺序，先遍历右子树并将节点入栈，然后更新节点值，最后遍历左子树。这种方法确保了累加树的正确构建。

摘要由CSDN通过智能技术生成

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode convertBST(TreeNode root) {
        //栈用于控制遍历的顺序
        Stack<TreeNode> stack = new Stack<TreeNode>();
        //当前节点
        TreeNode cur = root;
        //用于表示所有大于等于当前节点的值
        int sum = 0;
        //遍历
        while(!stack.isEmpty() || cur != null){
            //入栈又有右节点
            while(cur != null){
                stack.push(cur);
                cur = cur.right;
            }
            //出栈，更新sum重新赋值本节点
            cur = stack.pop();
            sum += cur.val;
            cur.val = sum;
            //继续遍历
            cur = cur.left;
        }
        return root;
    }
}