剑指Offer-26. 树的子结构

题目描述

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

思路

需要注意在第二步判断时，递归是否有 结束条件，以及多处判断一个 指针是否为 null。

此题第二步判断的递归结束条件是：root2 已经遍历完了，则说明存在子结构；或者 root1 已经遍历完了，则说明不存在子结构。

/**
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) {
        boolean res = false;

        if (root1 != null && root2 != null) {
            // 第一步判断
            // root1 中一个结点的值和 root2 根结点相同，则继续判断
            if (root1.val == root2.val)
                res = fun(root1, root2);
            if (!res)
                res = HasSubtree(root1.left, root2);
            if (!res)
                res = HasSubtree(root1.right, root2);
        }

        return res;
    }

    // 第二步判断，以 root1 中某结点为根的树是否和 root2 有相同结构
    public boolean fun(TreeNode root1, TreeNode root2) {
        // 如果 root2 先结束，说明遍历完了
        if (root2 == null)
            return true;
        // 如果 root1 先结束，说明 root2 不是其子结构
        if (root1 == null)
            return false;
        // 如果 root1 的该结点值和 root2 根结点不相等，肯定不是子结构
        if (root1.val != root2.val)
            return false;
        return fun(root1.left, root2.left) && fun(root1.right, root2.right);
    }
}