【效率百分百】剑指 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

---

解法：递归匹配

代码

/* 判断B是否为A的子树 */
public boolean isSubStructure(TreeNode A, TreeNode B) {
    if(A == null || B == null) return false;
    // B的根节点就是A的根节点，或者B的根节点在A的左右子树
    return isSub(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
}

/* 判断两个数是否匹配 */
public boolean isSub(TreeNode a, TreeNode b) {
    // b被匹配完了，当然是
    if(b == null) return true;
    // a被先匹配完了或者值不相同，当然不是
    if(a == null || a.val != b.val) return false;
    // 若当前结点值相同，还需要比较其左右子结点
    return isSub(a.left, b.left) && isSub(a.right, b.right);
}

提交结果