程序员面试金典——面试题 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

二、解题思路

       本题，分为以下几种情况：

（1）t2为NULL时，则t2为任意树的子树。

（2）t1为NULL且t2不为空时，t2不是t1的子树。

（3）当t1和t2均不为空，且t1 != t2时，在t1的左右子树中查找与t2的值相等的节点。（采用递归的方式）

（4）当t1和t2的值相等时，则t1和t2的所有子节点的值全都相等，才能满足t2为t1子树的条件。

三、解题代码

class Solution {
private:
    bool isFirst = true;
public:
    bool checkSubTree(TreeNode* t1, TreeNode* t2) {
        if(t2 == NULL) return true;
        if(t1 == NULL) return false;
        return dfs(t1, t2);
    }

    bool dfs(TreeNode* t1, TreeNode* t2)
    {
        if(t2 == NULL && t1 == NULL)
        {
            return true;
        }
        else if(t2 == NULL || t1 == NULL)
            return false;
        if(t1->val != t2->val)
            return dfs(t1->left, t2) || dfs(t1->right, t2);
        else
            return dfs(t1->left, t2->left) && dfs(t1->right, t2->right);
    }

};

四、解题结果