当前位置: 代码迷 >> 综合 >> Leetcode每日一题(20200921,二叉树)
  详细解决方案

Leetcode每日一题(20200921,二叉树)

热度:47   发布时间:2024-02-21 02:02:24.0

今日题目

T538 把二叉搜索树转换为累加树(简单,二叉树遍历)

题目描述

给定一个二叉搜索树(Binary Search Tree),把它转换成为累加树(Greater Tree),使得每个节点的值是原来的节点值加上所有大于它的节点值之和。

例如:

输入: 原始二叉搜索树:5/   \2     13输出: 转换为累加树:18/    \20     13

标签

二叉树

解析

二叉搜索树的中序遍历,遍历的顺序是值从小到大。将这个顺序倒转,我们就可以得到从大到小的遍历顺序。利用全局变量表示遍历到某个节点时,大于这个节点的节点值之和,就可以实现累加的效果。

python解法

#利用函数传值的暴力解法
class Solution:def convertBST(self, root: TreeNode) -> TreeNode:def calc(root, value):if not root:return 0rightVal = calc(root.right, value)leftVal = calc(root.left, value + root.val + rightVal)rootVal = root.valroot.val += rightValreturn rootVal + leftVal + rightValcalc(root, 0)return root
#逆向中序遍历
class Solution:def convertBST(self, root: TreeNode) -> TreeNode:sum = 0def reversedInorderReversal(root):nonlocal sumif not root:returnreversedInorderReversal(root.right)sum += root.valroot.val = sumreversedInorderReversal(root.left)reversedInorderReversal(root)return root