图解LeetCode——剑指 Offer 26. 树的子结构

一、题目

输入两棵二叉树A和B，判断B是不是A的子结构。(约定空树不是任意一个树的子结构) B是A的子结构， 即：A中有出现和B相同的结构和节点值。

二、示例

2.1> 示例 1：

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

2.2> 示例 2：

【输入】A = [3,4,5,1,2], B = [4,1]
【输出】true

限制：

• 0 <= 节点个数 <= 10000

三、解题思路

根据题目描述，我们需要判断二叉树B是否是二叉树A树的子结构，因为二叉树B不确定会与二叉树A的哪一部分完全匹配，那么我们就需要做到如下两步：

【步骤1】寻找二叉树A哪一个节点与二叉树B的root节点值相同。
【步骤2】满足第1个步骤之后，再对比二叉树B的其他节点，如果完全匹配则返回true，否则，再重复步骤1，直到二叉树A的所有节点都对比完毕。

那么针对步骤1，我们可以通过递归isSubStructure(TreeNode A, TreeNode B)方法来实现遍历二叉树A。即：

【case1】对比A与B；
【case2】对比A.left与B；
【case3】对比A.right与B；

针对步骤2，我们实现一个对比函数compare(TreeNode A, TreeNode B)，来判断两个二叉树每个节点是否相同，具体判断逻辑如下所示：

【case1】如果节点B等于null，表示二叉树B已经遍历完毕，返回true；
【case2】如果节点A等于null但是节点B不等于null，表示树B还没有对比完毕，而树A却遍历完毕了，则返回false；
【case3】递归调用compare(A.left, B.left) 和 compare(A.right, B.right);

解题思路如下所示，我们以【输入】A = [1,1,3,2,3,5,6], B = [1,2,3]为例，演示一下具体的对比流程。

四、代码实现

class Solution {
    public boolean isSubStructure(TreeNode A, TreeNode B) {
        if (A == null || B == null) return false;
        return compare(A, B) || isSubStructure(A.left, B) || isSubStructure(A.right, B);
    }
    /** 对比是否相同树节点 */ 
    public boolean compare(TreeNode A, TreeNode B) {
        if (B == null) return true; // B树对比完毕
        if (A == null || A.val != B.val) return false; // A树遍历完毕了，但是B树还存在未对比的节点
        return compare(A.left, B.left) && compare(A.right, B.right);
    }
}

 今天的文章内容就这些了：

写作不易，笔者几个小时甚至数天完成的一篇文章，只愿换来您几秒钟的 点赞 & 分享 。

更多技术干货，欢迎大家关注公众号“爪哇缪斯” ~ \(^o^)/ ~ 「干货分享，每天更新」