题目:输入两棵二叉树 A 和 B,判断 B 是不是 A 的子结构。
二叉树结点的定义:
/**
* 二叉树的树结点
*/
public static class BinaryTreeNode {
int value;
BinaryTreeNode left;
BinaryTreeNode right;
}
测试用例
- 功能测试(树A和树B都是普通二叉树,树B是或者不是树A的子结构)
- 特殊输入测试(两棵二叉树的一个或者两个根结点为null,二叉树的所有结点都没有左子树和右子树)
解题思路
要查找树 A 中是否存在和树 B 结构一样的子树,我们可以分成两步: 第一步在树 A 中找到和 B 的根结点的值一样的结点 R, 第二步再判断树 A 中以 R 为根结点的子树是不是包含和树 B 一样的结构。
我们来进一步分析。
第一步在树A中查找与根结点的值一样的结点,这实际上就是树的遍历。对二叉树这种数据结构熟悉的人当然知道可以用递归的方法遍历,也可以用循环。由于递归的代码实现起来比较简洁,面试的时候如果没有特别要求的话,通常会采用递归的方式。
/**
* 输入两棵二叉树A和B,判断B是不是A的子结构。
* 该方法是在A树中找到一个与B树的根节点相等的元素的结点,
* 从这个相等的结点开始判断树B是不是树A的子结构,如果找到其的一个就返回,
* 否则直到所有的结点都找完为止。
*
* @param root1 树A的根结点
* @param root2 树B的根结点
* @return true:树B是树A的子结构,false:树B是不树A的子结构
*/
public static boolean hasSubtree(BinaryTreeNode root1, BinaryTreeNode root2) {
// 只要两个对象是同一个就返回true
// 【注意此处与书本上的不同,书本上的没有这一步】
if (root1 == root2) {
return true;
}
// 只要树B的根结点点为空就返回true
if (root2 == null) {
return true;
}
// 树B的根结点不为空,如果树A的根结点为空就返回false
if (root1 == null) {
return false;
}
// 记录匹配结果
boolean result = false;
// 如果结点的值相等就,调用匹配方法
if (root1.value == root2.value) {
result = match(root1, root2);
}
// 如果匹配就直接返回结果
if (result) {
return true;
}
// 如果不匹配就找树A的左子结点和右子结点进行判断
return hasSubtree(root1.left, root2) || hasSubtree(root1.right, root2);
}
上述代码中,递归调用了hasSubtree的方法遍历二叉树A。如果发现某一结点的值和树B的头结点的值相同,就调用match方法做第二步判断,看看到底该结点对应的树是否和树B相同。
第二步是判断树A中以R为跟结点的子树是不是和树B具有相同的结构。同样,其实也是可以用递归的思路去考虑:如果结点R的值和树B的跟结点不同,那么这两个毫无疑问肯定不想等;如果它们的值相同,就递归判断它们各自的左右结点的值是不是相同。递归的终止条件就是我们到达了树A或者树B的叶结点。代码如下:
/**
* 从树A根结点root1和树B根结点root2开始,一个一个元素进行判断,判断B是不是A的子结构
*
* @param root1 树A开始匹配的根结点
* @param root2 树B开始匹配的根结点
* @return 树B是树A的子结构,false:树B是不树A的子结构
*/
public static boolean match(BinaryTreeNode root1, BinaryTreeNode root2) {
// 只要两个对象是同一个就返回true
if (root1 == root2) {
return true;
}
// 只要树B的根结点点为空就返回true
if (root2 == null) {
return true;
}
// 树B的根结点不为空,如果树A的根结点为空就返回false
if (root1 == null) {
return false;
}
// 如果两个结点的值相等,则分别判断其左子结点和右子结点
if (root1.value == root2.value) {
return match(root1.left, root2.left) && match(root1.right, root2.right);
}
// 结点值不相等返回false
return false;
}