剑指offer 26 树的子结构

@author: sdubrz
@date: 7/25/2020 4:03:31 PM   
难度： 中等
考察内容： 树
@e-mail: lwyz521604#163.com
题目来自《剑指offer》 电子工业出版社

输入两棵二叉树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

通过次数29,795提交次数64,231

解法

这道题做的时候需要搞清楚是判断树的子结构还是树的子树。这里是子结构，也就是说判断A中有没有一颗子树包含B，并且这颗子树可能有不属于B的其他叶子节点。

这道题我的思路是首先在A中寻找与B的根节点值相等的节点，然后在判断以这个节点为根的子树中是否包含B。下面是具体的Java实现：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(B==null || A==null){
            return false;
        }
        
        if(A.val==B.val){
            boolean find = isSubStructureRoot(A, B);
            if(find){
                return true;
            }
        }

        return isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }

    // A和B树根相同的情况下，A是否含有B
    public boolean isSubStructureRoot(TreeNode A, TreeNode B){
        if(A==null && B!=null){
            return false;
        }
        if(B==null){
            return true;
        }
        if(A.val!=B.val){
            return false;
        }

        boolean a = isSubStructureRoot(A.left, B.left);
        if(!a){
            return false;
        }
        boolean b = isSubStructureRoot(A.right, B.right);
        if(!b){
            return false;
        }

        return true;
    }
}

在 LeetCode 系统中提交的结果为

执行结果：通过 显示详情
执行用时：0 ms, 在所有 Java 提交中击败了100.00%的用户
内存消耗：41.5 MB, 在所有 Java 提交中击败了100.00%的用户