LeetCode之572. 另一个树的子树

题目

给定两个非空二叉树 s 和 t，检验 s 中是否包含和 t 具有相同结构和节点值的子树。s 的一个子树包括 s 的一个节点和这个节点的所有子孙。s 也可以看做它自身的一棵子树。

示例 1:
给定的树 s:

     3
    / \
   4   5
  / \
 1   2

给定的树 t：

   4 
  / \
 1   2

返回 true，因为 t 与 s 的一个子树拥有相同的结构和节点值。

示例 2:
给定的树 s：

     3
    / \
   4   5
  / \
 1   2
    /
   0

给定的树 t：

   4
  / \
 1   2

返回 false。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subtree-of-another-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

分析

这是一道简单题，但是没有直接AC，用例在90%左右的时候执行出错。

未看题解前

一开始看到题目，还以为是个简单的遍历二叉树的题。然后仔细往下看了看，发现并不是这么简单。

• 比较2棵树是否一致，可以采用前序遍历法对两棵二叉树进行遍历，每次递归都比较一下两棵树的val。
• 之前做过一道题，要求判断t是否属于s，即无论s下是否还存在子节点，t都属于s，即做以下判断：

if (s == nullptr && t == nullptr) {
	return true;
}

• 但是这里并没这么简单，需要的是s的子节点的所有子孙节点 == t，所以在这需要加上一层判断:

if (s == nullptr && t == nullptr) {
	return true;
}

// 新加判断
if (s == nullptr || t == nullptr) {
	return false;
}

• 递归遍历该二叉树。
• 以上分析会遗漏一个场景，当t已经遍历到了中间点时，当前的s节点若与t的根节点一致，则无法判断当前s节点是否是符合条件的子树 代码如下：
  写完以后发现自己想的有有些繁琐，应该不是正解，遂看了题解。

class Solution {
public:
    bool result = false;
    TreeNode *t1;
    bool isSubtree(TreeNode* s, TreeNode* t) {
        t1 = t;
        return dis(s, t);
    }

    bool dis(TreeNode* s, TreeNode* t) {
        if (result) {
            return result;
        }

        if (s == nullptr && t == nullptr) {
            return true;
        }

        if (s == nullptr || t == nullptr) {
            return false;
        }

        if (s->val == t->val) {
            bool ltmp = dis(s->left, t->left);
            bool rtmp = dis(s->right, t->right);
            if (t == t1 && ltmp && rtmp) {
                result = true;
            }
            return result ? result : ltmp && rtmp;
        } else {
            bool ltmp = dis(s->left, t);
            bool rtmp = dis(s->right, t);
            if (t == t1 && ltmp && rtmp) {
                result = true;
            }
            return result ? result : ltmp && rtmp;
        }
    }
};

看了题解后

二叉树问题不一定要想的那么复杂，如果想复杂了，建议重头开始整理思路。

看了题解后，发现自己想的太复杂了。对于二叉树的结构只需要明确一点：一个二叉树由其左子树和右子树构成。需要判断是否存在子树，只需要判断以下三点：

• t是否存在s的左子树内
• t是否存在s的右子树内
• t是否等于s

明确了以上3个条件后，就很容易写代码了。

代码

class Solution {
public:
    bool isSubtree(TreeNode* s, TreeNode* t) {
        // 加这一层判断防止下面递归访问非法内存
        if (s == nullptr) {
            return false;
        }
        return isSameTree(s, t) || isSubtree(s->left, t) || isSubtree(s->right, t);
    }

    bool isSameTree(TreeNode *s, TreeNode *t) {
        if (s == nullptr && t == nullptr) {
            return true;
        }

        if (s == nullptr || t == nullptr || s->val != t->val) {
             return false;
        }
        return isSameTree(s->left, t->left) && isSameTree(s->right, t->right);
    }
};