Medium | 剑指 Offer 26. 树的子结构 | 递归

剑指 Offer 26. 树的子结构

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

解体思路

比较A, B两棵树, 首先比较当前根节点是否相等, 然后比较左子树, 右子树, 如果根节点, 左右子树都相等则返回true。由于要判断B是否是A的子结构, 那么B节点为NULL, A不为NULL也是可以的。但是反过来就不行。

如果A当前根节点和B根节点比对, 比对不上, 则用A的左子树与B 比较, 否则再将A的右子树与B比较。

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

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