剑指 Offer 26. 树的子结构 Java版 题目及思路分析

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

思路分析：

要想判断B树是否是A树的子结构，分三种情况考虑：

（1）B树与A树相同。此时节点值相同，左子树结构相同，右子树结构相同。

（2）A的左子树中有与B相同的结构

（3）A的右子树中有与B相同的结构

这三种情况为或的关系。

使用递归函数完成。

public boolean isSubStructure(TreeNode A, TreeNode B) {
        //当A=null时，不能包含任何子结构，当B为null时，题目说明空树不是任意一个数的子结构，所以B也不能为null
		if(A == null || B == null)       return false;
        //分3种情况判断，采用helper函数递归判断第一种情况
		return helper(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
}

public boolean helper(TreeNode A, TreeNode B){
        //B为空是递归的终止条件
        if(B == null)        return true;
        //B不为null，A仍为null，返回false，否则报空指针异常
		if(A == null)        return false;
        return A.val == B.val && helper(A.left, B.left) && helper(A.right, B.right);
}