leetcode 面试题 04.10. 检查子树

检查子树。你有两棵非常大的二叉树:T1,有几万个节点;T2,有几万个节点。设计一个算法,判断 T2 是否为 T1 的子树。

如果 T1 有这么一个节点 n,其子树与 T2 一模一样,则 T2 为 T1 的子树,也就是说,从节点 n 处把树砍断,得到的树与 T2 完全相同。

注意:此题相对书上原题略有改动。

示例1:

输入:t1 = [1, 2, 3], t2 = [2]
输出:true
示例2:

输入:t1 = [1, null, 2, 4], t2 = [3, 2]
输出:false
提示:

树的节点数目范围为[0, 20000]。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/check-subtree-lcci
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

    public boolean checkSubTree(TreeNode A, TreeNode B) {
        if (B == null) {
            return true;
        }
        if (A == null) {
            return false;
        }
        return find(A, B) ;
    }

    public boolean find(TreeNode A, TreeNode B) {
        if (A == null) {
            return false;
        }
        return check(A, B) || find(A.left, B) || find(A.right, B);
    }

    private boolean check(TreeNode A, TreeNode B) {
        if (B == null && A == null) {
            return true;
        }
        if (B == null || A == null) {
            return false;
        }
        return A.val == B.val && check(A.left, B.left) && check(A.right, B.right);
    }

posted @ 2021-05-13 14:58  旺仔古李  阅读(58)  评论(0)    收藏  举报