用 Python 编写的程序，通过使用左子树和右子树本身的值来更新值，从而找到一棵树

假设我们有一棵二叉树，我们必须找到相同的树，但每个节点的值都被其值 + 其左子树和右子树的所有总和所替换。

因此，如果输入如下

那么输出将是

为了解决这个问题，我们将遵循以下步骤 −

• 定义一个函数 tree_sum() 。这将获取树的根

• 如果 root 为空，则

  • 返回 0

• 根的数据 := tree_sum(根的左侧) + tree_sum(根的右侧) + 根的数据

• 返回根的数据

• 从主方法中，执行以下操作:

• tree_sum(root)

• 返回根

让我们看看下面的实现以便更好地理解 −

示例

class TreeNode:
   def __init__(self, data, left = None, right = None):
      self.data = data
      self.left = left
      self.right = right

def inorder(root):
   if root:
      inorder(root.left)
      print(root.data, end = ', ')
      inorder(root.right)
   
class Solution:
   def solve(self, root):
      def tree_sum(root: TreeNode):
         if root is None:
            return 0
         root.data = tree_sum(root.left) + tree_sum(root.right) + root.data
return root.data
tree_sum(root)
return root
ob = Solution()
root = TreeNode(2)
root.left = TreeNode(3)
root.right = TreeNode(4)
root.left.left = TreeNode(9)
root.left.right = TreeNode(7)
ob.solve(root)
inorder(root)

输入

root = TreeNode(12)
root.left = TreeNode(8)
root.right = TreeNode(15)
root.left.left = TreeNode(3)
root.left.right = TreeNode(10)

输出

9, 19, 7, 25, 4,

相关文章

• 用 Python 编写程序，求出排序列表中每对数字的绝对差之和
• 用 Python 编写程序，查找使列表元素相等的最小总成本
• 用 Python 编写的程序，通过使用左子树和右子树本身的值来更新值，从而找到一棵树
• 用 Python 中的字符串字符计算我们可以生成的唯一回文数的程序
• 用 Python 计算顶点到顶点可达性矩阵的程序
• 用 Python 编写程序来计算投掷 n 个骰子的方法数
• 使用 C++ 对矩阵的每个对角线元素按升序排序的程序
• 用 Python 编写的程序，用于查找赢得删除重复整数游戏所需的移动次数
• 用 Python 编写一个程序，通过从第一个或最后一个字符中删除并再次插入来制作目标
• 用 Python 编写的程序，用于从二叉树中删除所有具有偶数值的叶子