一、题目
输入两棵二叉树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^)/ ~ 「干货分享,每天更新」