树的子结构

题目描述
输入两棵二叉树A，B，判断B是不是A的子结构。（ps：我们约定空树不是任意一个树的子结构）

我的Code如下：

/**
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) {
        if(root1 == null || root2 == null){
            return false;
        }
        boolean flag = false;
        if(root1.val == root2.val){
            flag =  tree1HasTree2(root1, root2);
        }
        //如果flag还为false的话，就去判断左子树和root2的关系
        if(!flag){
            flag = HasSubtree(root1.left, root2);
        }
        //如果flag还为false的话，就去判读右子树和root2的关系
        if(!flag){
            flag = HasSubtree(root1.right, root2);
        }
        return flag;
    }
    
    //该方法判断tree2是否存在于tree1中
    public boolean tree1HasTree2(TreeNode root1, TreeNode root2){
        //如果root2遍历完了，都没有返回false，说明tree2存在于tree1中
        if(root2 == null){
            return true;
        }
        //如果tree1遍历完了，tree2还没遍历完，说明tree2包含tree1，返回false
        if(root1 == null){
            return false;
        }
        //如果发现有值不相等的情况，直接返回false；
        if(root1.val != root2.val){
            return false;
        }
        //继续判断左子树和右子树是否对应上，
        return tree1HasTree2(root1.left, root2.left) && tree1HasTree2(root1.right, root2.right);
    }
}