剑指offer-26 树的子结构
微信搜索【程序员画工师】关注更多Java编程技术、数据结构与算法、面试题相关内容。
题目
输入两棵二叉树A,B,判断B是不是A的子结构。
思路
首先在树A中找到值为8(树B的根节点的值)的节点。从树A的根节点开始遍历,它的根节点的值就是8。接着判断树A的根节点下面的子树是不是和树B一样的结构。树A根节点的左子节点的值是8,而树B的根节点的左子节点是9,是不同的。
那再接着遍历树A,查找值为8的节点。在树的第二层找到了一个值为8的节点,然后进行第二步判断,于是遍历这个节点下面的子树,得到了两个子节点9和2,和树B的结构完全相同,因此树B是树A的子结构。如下图所示。
这道题需要两个递归,一个递归用来遍历在树A中和树B的根节点的值一样的节点R,一个递归用来判断以R为根节点的子树是不是和树B具有相同的结构。
上代码
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean isHave = false;
if(root1 == null){
return isHave;
}
if(root2 == null){
return isHave;
}
if(root1.val == root2.val){
isHave = judgeIsHave(root1,root2);
if(!isHave){
isHave = HasSubtree(root1.left,root2);
}
if(!isHave){
isHave = HasSubtree(root1.right,root2);
}
}
return isHave;
}
public boolean judgeIsHave(TreeNode root1,TreeNode root2){
if(root2 == null){
return true;
}
if(root1 == null){
return false;
}
if(root1.val != root2.val){
return false;
}
return judgeIsHave(root1.left,root2.left) && judgeIsHave(root1.right,root2.right);
}
}
思考
上述代码有多处判断节点是不是为空,这样做是为了避免试图访问空节点而造成的程序崩溃,同时设置了递归调用的退出条件。
在设置递归调用的退出条件时,judgeIsHave方法中判断小树是否为null必须放在前面,不能和判断大树是否为null交换,因为如果换了的话,在大树为null时,小树不确定,小树如果也为null,就应该是true,而这时返回的是false,就是错误的。
如果定义的二叉树的节点的类型为float或者double型时,由于计算机表示小数都有误差,不能直接用等号判断两个小数是否相等。如果两个小数的误差的绝对值很小,如小于0.0000001,就可以认为他们相等。参考代码如下:
public boolean equal(double num1,double num2){
if((num1 - num2 > -0.0000001) && (num1 - num2 < 0.0000001)){
return true;
}else{
return false;
}
}
References
[1] 《剑指offer(第二版)》 何海涛著
程序员画工师公众号,获取更多详细Java、Python、C、前端、小程序、产品相关学习资料,欢迎交流