剑指 Offer 26. 树的子结构

剑指 Offer 26. 树的子结构

题目

链接

https://leetcode-cn.com/problems/shu-de-zi-jie-gou-lcof/

问题描述

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

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

例如:
给定的树 A:

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

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

示例

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

提示

0 <= 节点个数 <= 10000

思路

首先判断根节点，之后开始比较从根节点，左节点，右节点是否包含子结构。

复杂度分析

时间复杂度 O(mn)
空间复杂度 O(1)

代码

Java

    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if(B == null || A == null) {
            return false;
        }
        return compare(A,B) || isSubStructure(A.left,B) || isSubStructure(A.right,B);
    }

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