题目描述:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
题解:
(1)我们规定,空树不是任何一个树的子结构,所以当 root1 或 root2 有一个为空时,就可以直接返回 false 。
(2)如果当前 root1.val == root2.val,那就可以去判断它的左子树和右子树是否相等,如果其左右子树都相等,那就继续判断左右子树的左右子树。依次类推,直至遍历结束。
(3)如果当前 root1.val 不等于 root2.val,那就看 root1.left.val 或者 root1.right.val 等不等于 root2.val,如果相等,那回到步骤2。直至 A树 遍历结束都没找到相等的结点,则说明 B树 不是 A树 的子结构。
public static boolean isSameTree(TreeNode root1,TreeNode root2){
// B树 遍历结束,说明在 A树 中找到了 B树
if(root2 == null){
return true;
}
// A树 遍历结束,说明在 A树 中没找到 B树
if(root1 == null){
return false;
}
return root1.val == root2.val
&& isSameTree (root1.left,root2.left)
&& isSameTree (root1.right,root2.right);
}
public static boolean HasSubtree(TreeNode root1,TreeNode root2){
if(root1 == null || root2 == null){
return false;
}
// 当前结点相等,判断其左右子树是否相等
if(root1.val == root2.val){
if(isSameTree (root1,root2)){
return true;
}
}
// root1 不等于 root2,就去看他的左右结点是否相等
return HasSubtree (root1.left,root2) || HasSubtree (root1.right,root2);
}