算法题：剑指 Offer 26. 树的子结构（题目+思路+代码+注释）时空 O(N^2) O(N) 0ms击败100%、22%用户

题目

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

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

示例 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

分解问题，要判断B树是不是A的子树，那B树的根节点一定是A树的某个节点，我们先遍历A树，找和B树根节点的点，然后拿这个点作为根节点的树去和B树比较，如果B树是该树的子树则直接返回true，否则继续找这样的节点，直到找完还没有则返回false。

代码

for遍历版本（慢于递归版本）

public boolean isSubStructure(TreeNode A, TreeNode B) {
        //题目约定：空树不是任何树的子结构
        if (B == null || A == null){
            return false;
        }
        //遍历A，找出起点相同的所有点
        Stack<TreeNode> stack1 = new Stack<>();
        stack1.push(A);
        //找值相同的点
        while (!stack1.isEmpty()){
            TreeNode t = stack1.pop();
            if (t.left != null){
                stack1.push(t.left);
            }
            if (t.right != null){
                stack1.push(t.right);
            }
            if (t.val == B.val && doJudgeSubstructure(t,B)){
                return true;
            }
        }
        return false;
    }
    
    public boolean doJudgeSubstructure(TreeNode A,TreeNode B){
        //出口  （刚进去B一定不为空）
        if (B == null){
            return true;
        }
        //A被遍历完了，B还没有结束
        if (A == null){
            return false;
        }
        if (A.val != B.val){
            return false;
        }else {
            return doJudgeSubstructure(A.left,B.left) && doJudgeSubstructure(A.right,B.right);
        }
    }

递归版本

public boolean isSubStructure(TreeNode A, TreeNode B) {
    return (A != null && B != null) && (doJudgeSubstructure(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B));
}

public boolean doJudgeSubstructure(TreeNode A, TreeNode B) {
   if (B == null) {
       return true;
   }
   if (A == null || A.val != B.val) {
       return false;
   }
   return doJudgeSubstructure(A.left, B.left) && doJudgeSubstructure(A.right, B.right);
}