树的子结构

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

爱上递归的我，这波却并不简洁：

import java.util.ArrayList;
/*
 * 思路：如果B是A的子树，首先满足B的值在A的结点中。
 * 因此，先用一个ArrayList<TreeNode> arr将与B的值相等的A中结点装起来
 * 再用一个函数遍历B树，同时检测从arr出发的子树是否和B相同
*/
 
public class Solution {
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root2==null||root1==null)return false;
        ArrayList<TreeNode> arr=new ArrayList<TreeNode>();
        arr=findSame(root1,root2,arr);
        for(TreeNode sameNode:arr)
        {
            if(isSubtree(sameNode,root2))
                return true;
        }
        return false;
    }
        //返回一个值与B相同的结点集合
    public ArrayList<TreeNode> findSame(TreeNode root1,TreeNode root2,ArrayList<TreeNode> arr){
        if(root1==null)
            return null;
        if(root1.val==root2.val)
            arr.add(root1);
        findSame(root1.left,root2,arr);
        findSame(root1.right,root2,arr);
        return arr;
    }
        //检测从该结点出发的树是否包含B树
    public boolean isSubtree(TreeNode sameNode,TreeNode root2)
    {
        if(root2!=null&&sameNode==null)
            return false;
        if(root2==null)//B树遍历完
            return true;
        if(root2.val==sameNode.val)
        {
            return isSubtree(sameNode.left,root2.left)&&isSubtree(sameNode.right,root2.right);
        }
        return false;
    }
}

看到的简洁版本：

public boolean IsSubtree(TreeNode root1, TreeNode root2){
        if(root2 == null)
            return true;
        if(root1 == null)
            return false;
        if(root1.val == root2.val)
        return IsSubtree(root1.left,root2.left) && IsSubtree(root1.right, root2.right);
        else return false;
    }
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        if(root1 == null || root2 == null)
            return false;
        //如果第一项不通过，则计算第二项、第三项，在这里递归
        return IsSubtree(root1, root2) || HasSubtree(root1.left, root2) || HasSubtree(root1.right, root2);
    }