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

遞歸設計思路：

A、判斷子樹根值是否相等。

A1、如果是相等，需要一個可以判斷該根左右孩子結點是否相等的遞歸函數。

A2、如果不等，找大樹的左右孩子結點當根結點，遞歸到A步驟。

# -*- 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):
        # write code here
        res = False
        if pRoot1 == None or pRoot2 == None :
            #根據題干要求返回False
            return res
        if pRoot1.val == pRoot2.val:
            #如果結點相等，Subtree函數負責判斷
            #這兩個結點的孩子是否相等
            res = Solution.Subtree(self, pRoot1, pRoot2)
        if not res:
            #當與小樹相同的子樹不在根部，找大樹左子樹
            res = Solution.HasSubtree(self, pRoot1.left, pRoot2)
        if not res:
            #左邊找到了就不需要再找了
            res = Solution.HasSubtree(self, pRoot1.right, pRoot2)
        return res
    
    def Subtree(self, pRoot1, pRoot2):
        #下面兩個if位置不能反
        if pRoot2 == None:
            return True
        if pRoot1 == None:
            return False
        if pRoot1.val != pRoot2.val:
            return False
        #左右同時相等時
        return Solution.Subtree(self, pRoot1.left, pRoot2.left) and \
    Solution.Subtree(self, pRoot1.right, pRoot2.right)        

 

為了簡化代碼，提高代碼的閱讀障礙，使用短路法

 

class Solution:
    def HasSubtree(self, pRoot1, pRoot2):
        # write code here
        if pRoot1 == None or pRoot2 == None :
            return False
        return Solution.Subtree(self, pRoot1, pRoot2) or \
    Solution.HasSubtree(self, pRoot1.left, pRoot2) or \
    Solution.HasSubtree(self, pRoot1.right, pRoot2)
    
    def Subtree(self, pRoot1, pRoot2):
        if pRoot2 == None:
            return True
        if pRoot1 == None:
            return False
        if pRoot1.val != pRoot2.val:
            return False
        return Solution.Subtree(self, pRoot1.left, pRoot2.left) and \
    Solution.Subtree(self, pRoot1.right, pRoot2.right)