树的子结构

题目：输入两颗二叉树A，B，判断B是不是A的子结构。

有两个任务，第一是比较B树和A的以某个根节点开始一个局部树是否完全相等（isSub()），第二是遍历A（HasSubtree()）。代码如下：

public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root1==null || root2==null) return false;
        if(root1.val==root2.val && isSub(root1, root2)) return true;
        else return (HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2));
    }
    // Compare only when root1.val == root2.val
    private boolean isSub(TreeNode root1, TreeNode root2) {
        if(root2.left==null && root2.right==null) return true;
        if(root2.left==null) { // root2.right not null
            if(root1.right==null) return false;
            if(root1.right.val!=root2.right.val) return false;
            return isSub(root1.right, root2.right);
        }
        if(root2.right==null) { // root2.left not null
            if(root1.left==null) return false;
            if(root1.left.val!=root2.left.val) return false;
            return isSub(root1.left, root2.left);
        }
        else { // root2.left not null and root2.right not null
            if(root1.left==null || root1.right==null) return false;
            if(root1.left.val!=root2.left.val || root1.right.val!=root2.right.val) return false;
            return (isSub(root1.left, root2.left) && isSub(root1.right, root2.right));
        }   
    }
}