输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
/**
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 ret = false;
//当root1和root2都不为空的时候,才进行比较
if (root1 != null && root2 != null) {
//在A树中如果找到了对应B的根节点 的 节点
if (root1.val == root2.val) {
//以这个节点为起点判断是否包含B树
ret = isSubtree(root1, root2);
}
//如果不包含,就去A树的左子树找
if (!ret) {
ret = HasSubtree(root1.left, root2);
}
//如果不包含,就去A树的右子树找
if (!ret) {
ret = HasSubtree(root1.right, root2);
}
}
return ret;
}
private boolean isSubtree(TreeNode root1, TreeNode root2) {
//如果B树已经遍历完,且A树都匹配返回true
if (root2 == null) {
return true;
}
//如果A树遍历完,B树没有遍历完
if (root1 == null) {
return false;
}
//有节点的值不相等
if (root1.val != root2.val) {
return false;
}
//根节点匹配,就去匹配其左右子树
return isSubtree(root1.left, root2.left) &&
isSubtree(root1.right, root2.right);
}
}