输入两棵二叉树A和B,判断B是不是A的子结构。(约定空树不是任意一个树的子结构)
B是A的子结构, 即 A中有出现和B相同的结构和节点值。
例如:
给定的树 A:
3
/ \
4 5
/ \
1 2
给定的树 B:
4
/
1
示例 1:
输入:A = [1,2,3], B = [3,1]
输出:false
示例 2:
输入:A = [3,4,5,1,2], B = [4,1]
输出:true
解题思路:
若B树是A树的子树,则子结构的根节点可能是A树的任一节点,所以要完成以下两个工作:
- 先序遍历树A中的每个节点(对应函数为
isSubStructure(A, B)
) - 判断树A中以每个节点的子树是否包含B(对应函数为
dfs(A,B)
)
dfs(A,B)
函数
1.终止条件
- 当节点B为空:说明已经完成了匹配,返回true;
- 当节点A为空:说明已经越过了树A叶子节点,返回false
- 当节点A.val与节点B.val不相等时,返回false
2.返回值
- 判断 A 和 B 的左子节点是否相等,即 recur(A.left, B.left);
- 判断 AA 和 BB 的右子节点是否相等,即 recur(A.right, B.right) ;
isSubStructure(A, B)
函数:
-
特例处理: 当 树 A 为空 或 树 B 为空 时,直接返回 falsefalse ;
-
返回值:若树 B 是树 A 的子结构,则必满足以下三种情况之一,因此用或 || 连接
- 以节点A为根节点的子树包含B,即isSubStructure(A,B)
- 树B是树A左子树的结构,即 isSubStructure(A.left,B)
- 树B是树A右子树的结构,即 isSubStructure(A.right,B)
其实也就对应着树A的前序遍历
代码:
class Solution {
public boolean isSubStructure(TreeNode A, TreeNode B) {
if(A == null || B ==null) return false;
return dfs(A,B) || isSubStructure(A.left,B) || isSubStructure(A.right,B);
}
public boolean dfs(TreeNode A,TreeNode B){
if(B==null) return true;
if(A==null || A.val !=B.val) return false;
return dfs(A.left,B.left) && dfs(A.right,B.right);
}
}