描述
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
示例1
输入: {8,8,#,9,#,2,#,5},{8,9,#,2}
返回值: true
思路
1.另起函数判断两个树是否相等,里面用到递归
2.分治法,先判断A是否和B相等,判断B是否是A的相同结构
3.如果不是,分别判断是否是A的左右子树的子结构
时间复杂度:O(MN),其中 M,N分别为树 pRoot1和 树 pRoot2的节点数量;遍历pRoot1:O(m),比较pRoot1和pRoot2:O(n)
空间复杂度:O(M),最差的情况就是遍历了pRoot1的所有节点
代码
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
boolean res = false;
if (root1 != null && root2 != null) {
if (root1.val == root2.val) {
//根节点val相等 判断A和B是否是相同的树
res = isSameTree(root1, root2);
}
if (!res) {
//如果不是 判断B是否是A左子树的子结构
res = HasSubtree(root1.left, root2);
}
if (!res) {
//如果不是 判断B是否是A右子树的子结构
res = HasSubtree(root1.right, root2);
}
}
return res;
}
public boolean isSameTree(TreeNode root1,TreeNode root2) {
if (root2 == null) {
return true;
}
if (root1 == null && root2 != null) {
return false;
}
if (root1.val != root2.val) {
return false;
}
return isSameTree(root1.left, root2.left)
&& isSameTree(root1.right, root2.right);
}
}