剑指offer-树的子结构C++实现

原题链接

#include <iostream>

using namespace std;

struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;

    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

class Solution {
public:
    bool HasSubtree(TreeNode *root1, TreeNode *root2) {
        // 若主树或子树有任意一个为空，那么必然不相同
        if (!root1 || !root2)return false;
        // 对称递归，先判断根节点是否相同
        // 若不同再递归向下判断左子树的根节点与子结构根节点是否相同
        // 若不同判断右子树的根节点与子结构根节点是否相同
        return recur(root1, root2) || HasSubtree(root1->left, root2) ||
               HasSubtree(root1->right, root2);
    }

    // 构建辅助函数 A为主树，B为子结构（注意这里不是子树）
    bool recur(TreeNode *A, TreeNode *B) {
        // 递归结束条件->A与B的对应位置上，B为空，说明B越过了叶节点，返回true
        if (!B)return true;
        // 递归结束条件->A与B的对应位置上，A为空，
        // 说明A越过叶节点，但是B没有越过，说明此时B不是A的子结构
        // 或者A与B对应的节点值不相等，此时返回false
        if (!A || A->val != B->val)return false;
        // 判断A与B的左子树是否相同，若相同再判断右子树是否相同
        return recur(A->left, B->left) && recur(A->right, B->right);
    }
};

思路：
这道题是判断子结构是否属于主树，很容易想到的方法是DFS(先序遍历)+包含判断。并且这是一道对称递归的题，又是判断子结构的问题，所以需要构建辅助函数。
对称递归指的是对一个对称的数据结构（比如二叉树），从整体的对称性来看，把大问题分解成子问题进行递归，不是单独考虑一部分（比如只考虑左子树或只考虑右子树），而是同时考虑对称的两部分（左右子树），比如这道题需要同时考虑子结构的左右子树与主树的左右子树是否相同，这类问题就是对称递归。
可以用对称性递归解决的二叉树问题大多是判断性问题(bool类型函数),这一类问题又可以分为以下两类：
1、不需要构造辅助函数。这一类题目有两种情况：第一种是单树问题，且不需要用到子树的某一部分(比如根节点左子树的右子树)，只要利用根节点左右子树的对称性即可进行递归。第二种是双树问题，即本身题目要求比较两棵树，那么不需要构造新函数。
2.需要构造辅助函数。这类题目通常只用根节点子树对称性无法完全解决问题，必须要用到子树的某一部分进行递归，即要调用辅助函数比较两个部分子树。形式上主函数参数列表只有一个根节点，辅助函数参数列表有两个节点。
参考一篇文章带你吃透对称性递归(思路分析+解题模板+案例解读)
时间复杂度O(MN)
其中 M,N 分别为树 A 和 树 B 的节点数量；先序遍历树 A占用O(M) ，每次调用 recur(A, B) 判断占用 O(N)。
空间复杂度O(M)
当树 A和树 B 都退化为链表时，递归调用深度最大。当M≤N 时，遍历树 A 与递归判断的总递归深度为 M ；当M>N 时，最差情况为遍历至树A 叶子节点，此时总递归深度为M。
参考树的子结构（先序遍历 + 包含判断，清晰图解）