剑指offer题解合集——Week3day3

剑指offerWeek3

周三：树的子结构

题目链接：树的子结构

输入两棵二叉树 A，B，判断 B 是不是 A的子结构。

我们规定空树不是任何树的子结构。

数据范围
每棵树的节点数量 [0,1000]

样例
树 A：

     8
    / \
   8   7
  / \
 9   2
    / \
   4   7
树 B：

   8
  / \
 9   2
返回 true，因为 B是 A的子结构。

AC代码

递归版本

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool hasSubtree(TreeNode* r1, TreeNode* r2) {
        if (!r1 || !r2) return false;
        if (isSame(r1, r2)) return true;
        
        return hasSubtree(r1->left, r2) || hasSubtree(r1->right, r2);
    }
    
    bool isSame(auto r1, auto r2)
    {
        if (!r2) return true;
        if (!r1 || r1->val != r2->val) return false;
        return isSame(r1->left, r2->left) && isSame(r1->right, r2->right);
    }
};

思路：

整体思路

简单模拟
创建一个虚拟头节点
当两个链表都存在之时，哪个链表的值小，就放到新链表里