题目描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解题思路
遍历A二叉树的每一个结点 如果A二叉树中以这个结点形成的子树 有任意一个和B子树结构一样 那么B就是A的子结构 我们可以用递归来求解
下面是代码实现
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(root2 == null || root1 == null) {
return false;
}
return NodeHasSubtree(root1,root2)|| HasSubtree(root1.left, root2) ||
HasSubtree(root1.right, root2);
}
public boolean NodeHasSubtree(TreeNode root1,TreeNode root2) {
if(root2 == null) {
return true;
}
if(root1 == null) {
return false;
}
if(root1.val == root2.val) {
return NodeHasSubtree(root1.left, root2.left) &&
NodeHasSubtree(root1.right, root2.right);
}
return false;
}
}
我们可以看到HasSubtree函数 所做的是 前序遍历每个结点 首先判断当前结点所形成的子树是否和B树一样 然后递归判断 左数和右树是否满足 条件 只要这3个情况有一个满足就代表B是A的子树。
那如何判断A树中以该节点为头节点的树是否和 B树相等呢
我们用了NodeHasSubtree函数来求解
如果A树的当前节点为空 B子树当前结点为空 return true
B子树当前节点不为空 return false
如果A树当前节点不为空 B子树当前节点为空 return true
B子树当前节点不为空 判断值是否相等 继续递归判断左子树右子树;
那么可以简化为