题目描述
输入两棵二叉树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
解题思路:
-
先序遍历树A中的每个节点(对应函数isSubStructure(A,B))
-
判断树A中的每个节点为根节点的子树是否包含树B。(对应函数isSub(A,B))
recur(A, B) 函数:
终止条件:
当节点 B 为空:说明树 B 已匹配完成(越过叶子节点),因此返回 true ;
当节点 A 为空:说明已经越过树 A 叶子节点,即匹配失败,返回 false ;
当节点 A 和 B 的值不同:说明匹配失败,返回 false ;
返回值:
判断 A 和 B 的左子节点是否相等,即 isSub(A.left, B.left) ;
判断 A 和 B 的右子节点是否相等,即 isSub(A.right, B.right) ;
isSubStructure(A, B) 函数:
特例处理: 当 树 A 为空 或 树 B 为空 时,直接返回 false ;
返回值: 若树 B 是树 A 的子结构,则必满足以下三种情况之一,因此用或 || 连接;
- 以 节点 A 为根节点的子树 包含树 B ,对应 isSub(A, B);
- 树 B 是 树 A 左子树 的子结构,对应 isSubStructure(A.left, B);
- 树 B 是 树 A 右子树 的子结构,对应 isSubStructure(A.right, B);
class Solution{
public boolean isSubStructure(TreeNode A, TreeNode B){
// 1.定义一个标志位flag用来做判断
boolean flag = false;
if (A != null && B != null) {
// 2.1 如果从根结点就找到,并且递归成功,返回true
flag = isSub(A, B);
// 2.2 如果根结点开始的没有找到,从根结点的左子结点开始搜索
if (!flag) {
flag = isSubStructure(A.left, B);
}
// 2.3 如果以上都没有找到,就从根结点的右子节点开始搜索
if (!flag) {
flag = isSubStructure(A.right,B);
}
}
// 3.最后返回flag即得到结果
return flag;
}
//函数DoesTree1HasTree2()实现判定A树是否包含B树结构的功能
boolean isSub(TreeNode A, TreeNode B){
// 1.首先判断B是否为空,一旦B为空,说明B已经递归到叶子结点
if (B == null) {
return true;
}
// 2.如果A == null,则返回false
if (A == null) {
return false;
}
// 3.如果A.val != B.val,返回false
if (A.val != B.val) {
return false;
}
// 4.如果以上情况都不是,说明A.val == B.val,我们继续向下递归判断
return isSub(A.left, B.left) && isSub(A.right, B.right);
}
}
内存优化后:
public boolean isSubStructure(TreeNode A, TreeNode B) {
//边界条件判断,如果A和B有一个为空,返回false
if (A == null || B == null)
return false;
return isSub(A, B);
}
boolean isSub(TreeNode A, TreeNode B) {
//这里如果B为空,说明B已经访问完了,确定是A的子结构
if (B == null)
return true;
//如果B不为空A为空,或者这两个节点值不同,说明B树不是
//A的子结构,直接返回false
if (A == null || A.val != B.val)
return false;
//当前节点比较完之后还要继续判断左右子节点
return isSub(A.left, B.left) && isSub(A.right, B.right);
}
参考链接:https://leetcode-cn.com/u/jyd/