一 题目
输入两颗二叉树A和B,判断B是不是A的子结构。二叉树节点的定义如下:
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
二 思路分析
解决这个问题,可以分为两步,第一步是在树A中找到和树B的根节点的值一样的节点R;第二步,判断树A中的根节点子树是不是包含和树B一样的结构。
2.1 第一步 在A中查找到和树B的根节点的值一样的节点R
从树A的根节点开始遍历,如果存在与树B的根节点的值一样的节点R,那么就继续判断R的子树是否和树B结构一致
2.2 第二步 树与树之间的包含关系
- 判断两个树根节点是否相同,
- 不相同,则直接返回false;
- 相同,继续判断左子树的根节点是否和B树的根节点值相同
- 直到,遇到A或者B的叶子节点的
- 如果A先到叶子节点,就说明在A中存在和B中相同的节点路径,但是,该路径上A要比B的短。如
树A中的相同路径是:8->7->2
树B中的相同路径是:8->7->2->6
虽然节点2之前都相同,但是A先到节点,所以还是要返回false - 如果B先到叶子节点,那么就表示存在相同的路径,返回true
- 此时,只有一条路径判断ok了,还有其他路径没有判断,所以,还需要判断其他的路径。
三 代码实现
3.1 树与树之间的包含关系
递归的思路可以用下面的过程描述
1.初始化算法。递归程序通常需要一个开始时使用的种子值(seed value)。要完成此任务,可以向函数传递参数,或者提供一个入口函数, 这个函数是非递归的,但可以为递归计算设置种子值。1
2.检查要处理的当前值是否已经与基线条件相匹配。如果匹配,则进行处理并返回值。
3.使用更小的或更简单的子问题(或多个子问题)来重新定义答案。
4.对子问题运行算法。
5.将结果合并入答案的表达式。
6.返回结果。
总结流程:初始化——>检查当前值与基线条件的匹配——>化小重定义——>对子问题运行算法——>结果归总——>返回结果
如果根节点的值相同,那么继续判断左子树和右子树。用同样的方法来解决更小或者更简单的子问题,可以用递归来解决这个问题。
public boolean doesTree1HaveTree2(TreeNode tree1, TreeNode tree2) {
// 被包含的树判断的相同路径先到达叶子节点,那么就可以返回true,说明该路径匹配成功
if (tree2 == null) {
return true;
}
// 如果tree1先到达叶子节点,说明在A中存在和B中相同的节点路径,但是,该路径上A要比B的短。
if (tree1 == null) {
return false;
}
if (tree1.val != tree2.val) {
return false;
}
return doseTree1HaveTree2(tree1.left, tree2.left) && doseTree1HaveTree2(tree1.right, tree2.right);
}
3.2 查找与B根节点值相同的节点
分析问题,首先判断A的根节点,如果不相同再判断A的左右子节点是否相同。
public boolean hasSubtree(TreeNode A, TreeNode B) {
boolean result = false;
if (A != null && B !=null) {
if (A.val == B.val) {
result = doseTree1HaveTree2(A, B);
}
// 如果树A的根节点的值和树B的根节点的值相同,但是在以A的根节点作为匹配路径的起点,没有找到与树B相同的子树,就判断A的左子节点的值与B的根节点值。
if (!result) {
result = hasSubtree(A.left, B);
}
if (!result) {
result = hasSubtree(A.right, B);
}
}
return result;
}