《程序员面试金典（第6版）》面试题 04.10. 检查子树-CSDN博客

本文链接：https://blog.csdn.net/weixin_49503250/article/details/129625976

文章讲述了如何设计一个算法来判断一棵二叉树是否是另一棵大树的子树，强调了正确处理边界条件和递归逻辑的重要性。通过错误和正确代码案例的分析，展示了将比较逻辑抽离成独立函数可以提高代码的清晰度和正确性。时间复杂度为O(M*N)，空间复杂度为O(H)。

摘要由CSDN通过智能技术生成

题目描述

检查子树。你有两棵非常大的二叉树：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]。

解题思路与代码

这道题难度为中等题，我觉得是很有道理的。因为在做这道题时，其实会有很多边界条件头脑去想的不够清晰，最后导致题解会发生错误。

那现在就由我来解开这道题神秘的面纱吧！

前序遍历（依次判断，中间节点，左子树，右子树）

这道题的题意算是不难理解，我们按照题意很清楚的明白它要让我们做什么。其实就是让我们判断两个树是否完全相等。如果不完全相等，那 t2 是否为 t1 的一颗子树。

那大致我们就可以抽象出如下的逻辑，首先对比两颗树节点的值是否相等，如果相等，再去递归对比它的左子树与右子树是否都相等。如果完全相等就返回true。

否则那就说明t1 与 t2不相等，那么再去判断当t1 等于它的左子树，或右子树时，是否与t2 完全相等。

若t2 等于t1的某颗子树，那么就返回true，否则返回false。

代码大致的逻辑就是这样。但即使是这样，我们写代码还是很容易写错。

容易错的点在于，你如果只是考虑t2是不是t1的子树，那你就陷入圈套了。其实这道题真正的意思是看你能不能将判断t1 是否与t2相等这个逻辑抽离出来。因为我们肯定是先判断，t1 与 t2 是否完全相等，再去判断 t1 的某棵子树是否与t2相等的。

分析错误案例

来我们大家来看一个错误的代码案例：

class Solution {
public:
    bool checkSubTree(TreeNode* t1, TreeNode* t2) {

        if(t1 == nullptr && t2 == nullptr) return true;
        if(t1 == nullptr || t2 == nullptr) return false;

        if(t1->val == t2->val){
            bool left = checkSubTree(t1->left,t2->left);
            bool right = checkSubTree(t1->right,t2->right);
            if(left && right) return true;
            return checkSubTree(t1->left,t2) || checkSubTree(t1->right,t2);
        }else{
            return checkSubTree(t1->left,t2) || checkSubTree(t1->right,t2);
        }
    }
};

看这个错误代码，我们就是没有把 t1 与 t2 是否完全相等的逻辑彻底判断完毕，就匆匆忙忙的进入了选择递归的判断。从而只考虑以该节点为根节点、左右子树与t2相同的情况。但是实际上它还可能是在t1中另一位置的某个子树上有节点与t2的根节点值相同。

分析正确案例

现在我们来写正确的代码：

我们将判断的逻辑从原函数中抽离出来。单独作为一个递归函数。这个函数的作用就是单纯的判断两个二叉树是否是相等的二叉树。
那这样就很简单了，在主函数里，我们只需要去看这个t1的节点的值与t2的节点值是否相等。如果相等，那就直接扔到判断的函数中去检查，这两个数是否完全相等。如果是，直接返回true。如果不是，再去递归判断t1的子树是否与t2完全相等就好啦。

具体的代码如下：

class Solution {
public:
    bool checkSubTree(TreeNode* t1, TreeNode* t2) {

        if(t1 == nullptr && t2 == nullptr) return true;
        if(t1 == nullptr || t2 == nullptr) return false;

        if (t1->val == t2->val && isEqual(t1, t2)) return true;
       
        return checkSubTree(t1->left, t2) || checkSubTree(t1->right, t2);  // 当前结点不匹配，继续递归左右子树
    }

    bool isEqual(TreeNode* t1, TreeNode* t2) {
        if(t1 == nullptr && t2 == nullptr) return true;
        if(t1 == nullptr || t2 == nullptr) return false;

        if (t1->val != t2->val) return false;
        return isEqual(t1->left, t2->left) && isEqual(t1->right, t2->right);
    }
   
};