Java输入两棵二叉树AB-判断B是不是A的子树

题目描述

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

/**
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;
        }
        
        //如果找到与子树相同根的值，走判断方法
        if(root1.val == root2.val){
            if(judge(root1,root2)){
                return true;
            }
        }
        //遍历左孩子，右孩子 继续寻找root1与子树相同根的值
        return HasSubtree(root1.left,root2) || HasSubtree(root1.right,root2);
    }
    
    //判断是否是子结构
    public boolean judge(TreeNode root1,TreeNode root2){
        
        //如果目标树各个节点已经遍历完成 说明 子树是目标树root1的子树
        if(root2 == null){
            return true;
        }
        
        //上面目标树 已经遍历完成了,此时如果目标树遍历未完成,root1树遍历完成了,那么说明root2不是root1的子树
        if(root1 == null){
            return false;
        }
        
        //相等后判断左右孩子
        if(root1.val == root2.val){
            return judge(root1.left,root2.left) && judge(root1.right,root2.right);
        }
        
        return false;
    }
    
    
}