输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
第一步:在树A中查找与根节点的值一样的结点,这实际上就是树的遍历
第二步:判断树A中以R为根节点的子树,是否和树B有相同的结构。
// 判断 root2 是否是 root1 的子结构
public boolean HasSubtree(TreeNode root1, TreeNode root2) {
boolean res = false;
// 只有当root1和root2都不为null时,才有判断的必要
if(root1 != null && root2 != null){
//如果root1和root2的val相等
if(root1.val == root2.val){
//遍历两棵树,比较root2中的节点是否都跟root1的值相同
res = tree1HasTree2(root1, root2);
}
//如果不相同
if(!res){
//判断root2是否在root1的左子树里
res = HasSubtree(root1.left, root2);
}
//还是不相同
if(!res){
//判断root2是否在root1的右子树里
res = HasSubtree(root1.right, root2);
}
}
return res;
}
//递归比较root2的值是否与root1对应的值相同
public boolean tree1HasTree2(TreeNode root1, TreeNode root2){
//root2为空,表明root2已经比较到叶子节点,返回
if(root2 == null){
return true;
}
//如果root1为空,表名两棵树不一样
if(root1 == null){
return false;
}
//如果根节点不一样,肯定两棵树不一样
if(root1.val != root2.val){
return false;
}
//比较左子树和右子树
return tree1HasTree2(root1.left, root2.left) && tree1HasTree2(root1.right, root2.right);
}