面试题26. 树的子结构

题目

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构)

B是A的子结构， 即 A中有出现和B相同的结构和节点值。

例如:

给定的树 A:

     3
    / \
   4   5
  / \
 1   2
给定的树 B：

   4 
  /
 1

返回 true，因为 B 与 A 的一个子树拥有相同的结构和节点值。

示例 1：

输入：A = [1,2,3], B = [3,1]
输出：false

示例 2：

输入：A = [3,4,5,1,2], B = [4,1]
输出：true

限制：

0 <= 节点个数 <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解法

细节题，思路很直白，先在A树遍历找到B树的根，然后再递归对比。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def isSubStructure(self, A: TreeNode, B: TreeNode) -> bool:
        def rec(A, B):
            # B树完了意味着找到了，A树完了意味着没找到
            if B is None: return True
            if A is None: return False
            if A.val == B.val and rec(A.left,B.left) and rec(A.right,B.right)  : return True
            else : return False
            

        if A is None or B is None:
            return False
        # 假定数字不重复
        if A.val == B.val: return rec(A, B)
        return self.isSubStructure(A.left,B) or self.isSubStructure(A.right,B)
   

总结

1. B树完了意味着找到了，A树完了意味着没找到
2. 假设数字不重复
3. 原函数就可以拿来做DFS，不用重新开一个。