剑指 Offer 26. 树的子结构

输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。

例如:
返回 true,因为 B 与 A 的一个子树拥有相同的结构和节点值。
示例 1:
输入:A = [3,4,5,1,2], B = [4,1]
输出:true

  1. //递归 时间Omn,空间On
  2. func isSubStructure(A *TreeNode, B *TreeNode) bool {
  3. if A == nil || B == nil {
  4. return false
  5. }
  6. if A.Val == B.Val {
  7. if isSon(A, B) == true {
  8. return true
  9. }
  10. }
  11. return isSubStructure(A.Left, B) || isSubStructure(A.Right, B) // 先比A的一部分和B
  12. }
  13. func isSon(A *TreeNode, B *TreeNode) bool {
  14. if B == nil {
  15. return true
  16. }
  17. if A == nil || A.Val != B.Val {
  18. return false
  19. }
  20. return isSon(A.Left, B.Left) && isSon(A.Right, B.Right)
  21. }