题目:输入两颗二叉树A和B,判断B是不是A的子结构。ps:空树不是任意一颗树的子结构。
例如下图两课树,B树就是A树的子树
思路:
第一步:需要找到A树中与B树的根节点相等的节点R;
第二步:判断A树中以R为根节点的子树中有没有和B树相等的;
具体实现:
我们定义一个函数实现遍历功能采用递归方式实现,如果节点不等于B树的根节点的值,继续进行遍历如果节点等于B树根节点的值调用一个实现步骤二的函数,该函数同样采用递归实现。
public class Solution {
public boolean HasSubtree(TreeNode root1,TreeNode root2) {
if(root1 == null || root2 == null) return false;
return hasSubtree(root1,root2);
}
public static boolean hasSubtree(TreeNode root1,TreeNode root2){
boolean result = false;
if(root1 != null && root2 != null){
if(root1.val == root2.val){
result = doesTree1HaveTree2(root1,root2);
if(!result)
result = hasSubtree(root1.left,root2);
if(!result)
result = hasSubtree(root1.right,root2);
}
}
return result;
}
public static boolean doesTree1HaveTree2(TreeNode root1,TreeNode root2){
if(root2 == null) return true;
if(root1 == null) return false;
if(root1.val != root2.val ) return false;
return doesTree1HaveTree2(root1.left,root2.left) && doesTree1HaveTree2(root1.right,root2.right);
}
}