Pythonで1つのツリーが他のツリーのサブツリーであるかどうかを確認するプログラム

2つの二分木があるとします。 2番目のツリーが最初のツリーのサブツリーであるかどうかを確認する必要があります。

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

その場合、出力はTrueになります。

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

• 関数solve（）を定義します。これはルート、ターゲットになります

• ルートがnullで、ターゲットもnullの場合、

  • Trueを返す

• ルートがnullまたはターゲットがnullの場合、

  • Falseを返す

• ルートの値がターゲットの値と同じである場合、

  • 戻り値solve（ルートの左、ターゲットの左）とsolve（ルートの右、ターゲットの右）

• それ以外の場合

  • 戻り値solve（ルートの左側、ターゲット）またはsolve（ルートの右側、ターゲット）

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

例

class TreeNode:
   def __init__(self, data, left = None, right = None):
      self.val = data
      self.left = left
      self.right = right
class Solution:
   def solve(self, root, target):
      if root == None and target == None:
         return True
      if root == None or target == None:
         return False
      if root.val == target.val:
         return self.solve(root.left, target.left) and
self.solve(root.right, target.right)
      else:
         return self.solve(root.left, target) or
self.solve(root.right, target)
ob = Solution()
root1 = TreeNode(6)
root1.left = TreeNode(4)
root1.right = TreeNode(10)
root1.left.left = TreeNode(3)
root1.left.right = TreeNode(5)
root2 = TreeNode(4)
root2.left = TreeNode(3)
root2.right = TreeNode(5)
print(ob.solve(root1, root2))

入力

root1 = TreeNode(6)
root1.left = TreeNode(4)
root1.right = TreeNode(10)
root1.left.left = TreeNode(3)
root1.left.right = TreeNode(5)
root2 = TreeNode(4)
root2.left = TreeNode(3)
root2.right = TreeNode(5)

出力

True