牛客网 树的子结构

题目：

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

分析：

要查找树A中是否存在和树B一样的子树，需要分成两步：

1、在树A中找到和树B的根节点的值一样的节点R，实际上就是树的遍历。

2、再判断树A中以R为根节点的子树是不是包含和树B一样的结构。

代码：

/**
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) {//在A中查找与B的根节点一样的节点
        boolean result=false;
        if(root1==null||root2==null){//存在空树
            return result;
        }
         if(root1!=null&&root2!=null){
             if(root1.val==root2.val){//如果根节点一样，就要再判断子树是否跟B的结构一样
                 result=DoesTree1HavaTree2(root1,root2);
             }
             if(!result){//如果根节点不一样，就要遍历左子树
                 result=HasSubtree(root1.left,root2);
             }
             if(!result){//如果根节点不一样，就要遍历右子树
                 result=HasSubtree(root1.right,root2);
              }
         }
        return result;
    }
    public static boolean DoesTree1HavaTree2(TreeNode root1,TreeNode root2){//用来以R为根节点的子树和B树有相同的子结构
        //如果root1已经遍历完了，就返回false
        if(root1==null&&root2!=null){
            return false;
        }
        //如果root2先遍历完，就返回true
        if(root2==null){
            return true;
        }
        if(root1.val!=root2.val){
            return false;
        }
        return DoesTree1HavaTree2(root1.left,root2.left) && DoesTree1HavaTree2(root1.right,root2.right);
    }
}