劍指offer32 --- 樹的子結構（Python）

劍指offer32 — 樹的子結構（Python）

題目

輸入兩棵二叉樹A，B，判斷B是不是A的子結構。（ps：我們約定空樹不是任意一個樹的子結構）

分析

先判斷root的情況（判斷root的情況時也是先判斷root，再判斷左右子樹是否一致（遞歸））, 再判斷左右子樹（遞歸）。具體見代碼。

解

# -*- coding:utf-8 -*-
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # 邊界條件
        if pRoot1==None or pRoot2==None:
            return False
        
        def HasEqual(pRoot1,pRoot2):
            if pRoot2 == None:
                return True
            if pRoot1 == None:
                return False
            
            if pRoot1.val == pRoot2.val:
                if pRoot2.left == None:
                    leftEqual = True
                else:
                    leftEqual = HasEqual(pRoot1.left,pRoot2.left)
                    
                if pRoot2.right == None:
                    rightEqual = True
                else:
                    rightEqual = HasEqual(pRoot1.right,pRoot2.right)
                return leftEqual and rightEqual #左右均相等的時候才返回True
            return False
        
        # 1. 判斷根結點的情況
        if pRoot1.val == pRoot2.val:
            ret = HasEqual(pRoot1,pRoot2) #遞歸判斷是否相等
            if ret:
                return True
            
        # 2. 判斷左子樹的情況
        ret = self.HasSubtree(pRoot1.left, pRoot2)       
        if ret:
            return True
        
        # 3. 判斷右子樹的情況
        ret = self.HasSubtree(pRoot1.right, pRoot2)
        if ret:
            return True

牛客網該題鏈接