题目:输入两棵二叉树A和B,判断B是不是A的子结构。要注意二叉树的值是double值。
思路:整个分为两步。第一步在树A中查找与B中的根节点的值一样的节点,这实际上就是树的遍历:第二步是判断树A中以R为根节点的子树是不是和树B具有相同的结构。在整个过程中需要时刻注意节点是否为空。除此之外,由于节点的值都是double类型的,不能直接使用==比较大小,在这里专门写了一个函数Isequal来判断两个double值是否相等。
代码:
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean result = false;
if (root2 != null && root1 != null) { //首先判断如果两个根节点都不为空
if(Isequal(root1.val , root2.val)){ //如果两个根节点的值相等,继续判断结构是否相同
result = doesTree1HaveTree2(root1,root2);
}
if (!result) //如果上一步的值返回false或者两个根节点的值不相等,则继续沿着二叉树遍历,直到找到和树B根节点相同的值
return HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
}
//如果两个根节点有一个为空,直接返回false
return result;
}
public boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
if (node2 == null) {
return true;
}
if (node1 == null) {
return false;
}
if (!(Isequal(root1.val , root2.val)) {
return false;
}
return doesTree1HaveTree2(node1.left,node2.left) &&
doesTree1HaveTree2(node1.right,node2.right);
}
//这个是用来判断浮点数是否相等的函数
public boolean Isequal(double num1,double num2){
if(num1-num2>-0.0000001)&&(num1-num2<0.0000001)) //当两个浮点数差的范围
return true;
else
return false;
}