题目:输入两棵二叉树A和B,判断B是不是A的子结构。
例如:下图中所示的两棵二叉树,由于A中有一部分子树的结构和B 是一样的,因此B是A的子结构。
要查找树A中是否存在和树B结构一样的子树,我们可以分两步:
第一步:在树A中找到和B的根节点的值一样的结点R;
第二步:再判断树A中以R为根结点的子树是不是包含和树B一样的结构。
以上面的两棵树为例来详细分析这个过程。首先我们试着在树A中找到值为8的结点。从树A的根节点开始遍历,我们发现它的根节点的值就是8,接着我们就去判断树A的根节点下面的子树是不是含有和树B一样的结构。在树A中,根节点的左子结点的值是8,而树B的根节点的左子结点是9,对应的两个结点不同。
因此我们仍然要遍历A,接着查找值为8的结点。我们在树的第二层中找到一个值为8 的结点,然后进行第二步的判断,即判断这个结点下面的子树是否含有和树B一样结构的子树。于是我们遍历这个结点下面的子树,先后得到两个子节点9和2,这和树B的结构完全相同。此时我们在树A中找到了一个和树B的结构一样的子树,因此树B和树A的子结构。
可以发现上诉的两步其实都可以采用递归的方式实现。
还需要注意的而是一定要检查边界条件,即检查空指针。当树A或树B为空的时候,定义相同的输出。如果没有检查并做响应的处理,程序非常容易崩溃,这是面试的时候非常忌讳的事情。
public class FindChildTree {
class TreeNode{
private int val = 0;
private TreeNode left = null;
private TreeNode right = null;
public TreeNode(int val){
this.val = val;
}
}
public boolean hasSubTree(TreeNode root1, TreeNode root2){
boolean result = false;
// 当Tree1和Tree2都不为空时才进行比较,否则直接返回false
if(root1 != null && root2 != null){
// 如果在Tree1中找到了对应得Tree2的根节点
if(root1.val == root2.val){
// 在Tree1中以和root2相等的这个节点为根节点开始遍历判断是否包含Tree2
result = doesTree1HaveTree2(root1, root2);
}
// 如果找不到,那么就再去root1的左儿子当根节点,判断是否包含Tree2
if(!result){
result = hasSubTree(root1.left, root2);
}
// 如果找不到,那么就再去root1的右儿子当根节点,判断是否包含Tree2
if(!result){
result = hasSubTree(root1.right, root2);
}
}
// 返回结果
return result;
}
// 判断Tree1中以root1为根节点的树中是否包含Tree2
public boolean doesTree1HaveTree2(TreeNode node1, TreeNode node2) {
// 如果Tree2都遍历完了,且值都能对应上,则说明在Tree1中找到了Tree2
if(node2 == null){
return true;
}
// 如果Tree2还没有遍历完,Tree1却遍历完了,则直接返回false
if(node1 == null){
return false;
}
// 如果其中有一个节点没有对应上,返回false
if(node1.val != node2.val){
return false;
}
// 如果根节点对应的上,那么就分别取左右子节点中进行匹配
return doesTree1HaveTree2(node1.left, node2.left) &&
doesTree1HaveTree2(node1.right, node2.right);
}
}
这里需要说明一个面试中有可能遇到的坑:
本题中的TreeNode值类型为int,所以可以使用“==”判断两者是否相等,但是如果val的类型为long时,就不能直接使用“==”去比较了。这是因为在计算机内表示小数时(包括:float和double)都有误差。判断连个小数是否相等,只能判断它们之差的绝对值是不是在一个很小的范围内。如果两个数相差很小,就可以认为它们相等,可以定义一个equal函数:
public boolean equal(double n1, double n2){
if((ni - n2 > -0.0000001) && (n1 - n2 < 0.0000001)){
return true;
}else{
return false;
}
}
如果两个小数的差的绝对值很小,如小于0.0000001,就可以认为它们相等。