剑指Offer第十七题;树的子结构,检查一个树是否为另一个树的子树
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
外递归函数:
进行条件判断,当两个树都不为空时,才有可能是子结构
当找到一个结点和子结构的根结点相同时,进入内递归函数检查其结点是否相同
内递归函数:
当子树为空时,说明已经遍历了子树的所有结点,并且全部相同,返回true
当父树为空时,能进入这个if说明子树还不是null,因此不是子结构,返回false
当子树结点和父树结点相等时,正常递归检查左子树和右子树
当当前结点已经不想等时,返回false
/**
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) {
res = doesTree1HasTree2(root1, root2);
}
if(!res) {
res = HasSubtree(root1.left, root2);
}
if(!res) {
res = HasSubtree(root1.right, root2);
}
}
return res;
}
private boolean doesTree1HasTree2(TreeNode root1, TreeNode root2) {
if(root2 == null) {
return true;
}
if(root1 == null) {
return false;
}
if(root1.val == root2.val) {
return doesTree1HasTree2(root1.left, root2.left)&&doesTree1HasTree2(root1.right, root2.right);
}
return false;
}
}