树的子结构

题目描述

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

Solution

/**
public class TreeNode {
    int val = 0;
    TreeNode left = null;
    TreeNode right = null;

    public TreeNode(int val) {
        this.val = val;

    }

}
*/
public class Solution {
    //分两步走
    //1 在树A中找到和树B的根节点的值一样的节点R;
    //2 判断树A中以R为根节点的子树是不是包含和树B一样的结构
    public boolean HasSubtree(TreeNode root1,TreeNode root2) {
        //因为空树不是任何树的子结构
        boolean result = false;
        if(root1 != null && root2 != null){
            if(root1.val == root2.val){
                result = helper(root1,root2);
            }
            //如果当前树root1的值不和树root2的值相等，则看当前树的左子树和root2的关系
            if(!result){
                result = HasSubtree(root1.left,root2);
            }
            //如果当前树root1的值不和树root2的值相等，则看当前述的右子树和root2的关系
            if(!result){
                result = HasSubtree(root1.right,root2);
            }
        }
        return result;
    }
    //判断以root为根节点的子树是不是包含和树root2一样的结构
    public boolean helper(TreeNode root1,TreeNode root2){
        if(root2 == null){
            return true;
        }
        if(root1 == null){
            return false;
        }
        if(root1.val != root2.val){
            return false;
        }
        //根节点相同时，考察左右子树是否相同
        return helper(root1.left,root2.left)&& helper(root1.right,root2.right);
    }
}