输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
要查找树A中是否存在和树B结构一样的子树,我们可以分成两步:
1.在树A中找到和B的根结点的值一样的结点R;
2.判断树A中以R为根结点的子树是不是包含和树B一样的结构。
public static boolean HasSubtree(TreeNode root1, TreeNode root2) {
if (root1 == null && root2 != null) return false;
if (root2 == null) return false;
boolean flag = false;
if (root1.val == root2.val) {
//如果根节点相同
flag = isSubtree(root1, root2);
}
if (!flag) {
//判断root2是不是root1的左子树的子结构
flag = HasSubtree(root1.left, root2);
}
if (!flag){
//判断root2是不是root1的右子树的子结构
flag = HasSubtree(root1.right,root2);
}
return flag;
}
private static boolean isSubtree(TreeNode root1, TreeNode root2) {
if (root1 == null && root2 != null) return false;
//当root2为空时说明root2是root1的子结构 退出循环
if (root2 == null) return true;
//当根节点不同时,返回false
if (root1.val != root2.val) return false;
else {
//根节点相同,继续判断
return isSubtree(root1.left, root2.left) && isSubtree(root1.right, root2.right);
}
}