Pythonでバイナリツリーのリーフノードと非リーフノードを検索するプログラム

二分木があるとすると、2つの数字のリストを見つける必要があります。最初の数字はツリー内の葉の数で、2番目の数字は非葉ノードの数です。

したがって、入力が次のような場合

3つのリーフと2つの非リーフノードがあるため、出力は（3、2）になります。

これを解決するには、次の手順に従います-

• nがnullの場合、
  • return（0、0）
• nの左側がnullで、nの右側がnullの場合、
  • return（1、0）
• 左：=ソルブ（nの左）
• right：=resolve（right of n）
• return（left [0] + right [0]、1 + left [1] + right [1]）

理解を深めるために、次の実装を見てみましょう-

例

class TreeNode:
   def __init__(self, data, left = None, right = None):
      self.val = data
      self.left = left
      self.right = right
class Solution:
   def solve(self, n):
      if not n:
         return 0, 0
      if not n.left and not n.right:
         return 1, 0
      left, right = self.solve(n.left), self.solve(n.right)
      return left[0] + right[0], 1 + left[1] + right[1]
ob = Solution()
root = TreeNode(6)
root.left = TreeNode(2)
root.right = TreeNode(6)
root.right.left = TreeNode(10)
root.right.right = TreeNode(2)
print(ob.solve(root))

入力

root = TreeNode(6)
root.left = TreeNode(2)
root.right = TreeNode(6)
root.right.left = TreeNode(10)
root.right.right = TreeNode(2)

出力

(3, 2)