剑指 Offer -- 树的子结构(十七)

树的子结构(十七)

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

代码(已在牛客上 AC)

这道题主要注意, t 是 s 的一部分也可以直接返回 true, 即代码中的 if (!t->left && !t->right && t->val == s->val) return true; (即 t 已经到叶子节点了). LeetCode 原题是要求完全匹配的.

举个例子:

 t:
    1
   / \
  2   3

 s:
    1
   / \
  2   3
     / \
    4   5

结果是返回 true 的. 要解决 LeetCode 的原题, 只需要把上面的那条 if 语句删去即可.

// 这道题和 leetcode 中的 572. Subtree of Another Tree 不同,
// 如果 s 的中间部分(未必要到叶子节点)的结构和 t 的结构相同, 那么
// 仍然返回 true.
class Solution {
public:
    bool HasSubtree(TreeNode* s, TreeNode* t) {
        if (!s || !t) return false;
        return isSame(s, t) || HasSubtree(s->left, t) || HasSubtree(s->right, t);
    }
private:
    // 判断两棵树是不是相同, 下面代码会判断:
    // [2, 1, 4, 4, 3] 与 [2, 1, 4] 两棵树是相同的.
    // 即只要 t 是 s 在一开始的部分是相同的即可, 不需要完全匹配,
    // 要完全匹配的话, 只需要删除函数中第二个 if 语句.
    bool isSame(TreeNode *s, TreeNode *t) {
        if (!s || !t) return (s == t);
        if (!t->left && !t->right && t->val == s->val) return true;
        return (s->val == t->val) && isSame(s->left, t->left) && isSame(s->right, t->right);
    }
};