树的子结构

题目描述

输入两棵二叉树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) {
        // 无论root1还是root2为空，都直接返回false
        if(root2==null || root1==null) return false;
        
        boolean result = false;
        // 以此为根节点，判断roo2是否为root1的子树
        if(root1.val == root2.val) {
            result = isSubTree(root1, root2);
        }
        // 若没有找到根节点，则继续查找左右子树
        if(!result) {
            result = HasSubtree(root1.left, root2);
        }
        if(!result) {
            result = HasSubtree(root1.right, root2);
        }
        return result;
    }
    
    public boolean isSubTree(TreeNode root1, TreeNode root2) {
        if(root2==null) return true; // root2遍历完，则返回true
        if(root1==null) return false; // 若root2未遍历完，而root1遍历完，则返回false
        
        if(root1.val == root2.val) // 若相等，则继续遍历左右节点是否相等
            return isSubTree(root1.left, root2.left) && isSubTree(root1.right, root2.right);
        
        return false; // 若不相等，直接返回false
    }
}