输入两棵二叉树 A,B,判断 B 是不是 A 的子结构。我们约定空树不是任意一个树的子结构
基本思路:
-
首先在 A 中找到与 B 的根结点的值一样的结点,可以使用递归遍历查找
-
找到的话,判断 A 中该结点对应的子树是否与 B 相等,相等返回 true,方法结束,否则返回 false,方法继续执行
-
继续在 A 中遍历查找与 B 的根结点的值一致的结点,找到的话重复第二步,如果不存在这样的结点,直接返回 false
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean result = false;
// 当 Tree1 和 Tree2 都不为 null 时,才进行比较。否则直接返回 false
if(root2 != null && root1 != null) {
// 如果找到了对应 Tree2 的根节点的点
if(root1.val == root2.val) {
// 以这个根节点为起点判断是否包含 Tree2
result = doesTree1HaveTree2(root1, root2);
}
// 如果找不到,那么就再去找 root 的左儿子当起点,判断是否包含 Tree2
if(!result) {
result = HasSubtree(root1.left, root2);
}
// 如果还找不到,那么就再去找 root 的右儿子当起点,去判断是否包含 Tree2
if(!result) {
result = HasSubtree(root1.right, root2);
}
}
// 返回结果
return result;
}
public boolean doesTree1HaveTree2(TreeNode node1,TreeNode node2) {
// 如果 Tree2 已经遍历完了都能对应的上,返回 true
// 注意这里必须先判断 node2,因为 Tree1 和 Tree2 是同步比较的,此时 node1 可能为 null,也可能不为 null
if(node2 == null) {
return true;
}
//如果 Tree2 还没有遍历完,Tree1 却遍历完了,返回false
if(node1 == null) {
return false;
}
//如果其中有一个点没有对应上,返回 false
if (node1.val != node2.val) {
return false;
}
// 如果根节点对应的上,那么就分别去子节点里面匹配
return doesTree1HaveTree2(node1.left,node2.left) && doesTree1HaveTree2(node1.right,node2.right);
}
}
相关文章
暂无评论...