剑指offer-17 -- 树的子结构 - C++

题目描述

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

题解

题目抽象：给2棵树A，树B，判断B是否是A的子结构。
子结构定义：树A和树B的根结点相等，并且树A的左子树和树B的左子树相等，树A的右子树和树B的右子树相等

方法：递归求解

第一步：
根据题意可知，需要一个函数判断树A和树B是否有相同的结构。显然是个递归程序。可考察递归程序3部曲。

1. 递归函数的功能：判断2个数是否有相同的结构，如果相同，返回true，否则返回false
2. 递归终止条件：

• 如果树B为空，返回true，此时，不管树A是否为空，都为true
• 否则，如果树B不为空，但是树A为空，返回false，此时B还没空但A空了，显然false

1. 下一步递归参数：

• 如果A的根节点和B的根节点不相等，直接返回false
• 否则，相等，就继续判断A的左子树和B的左子树，A的右子树和B的右子树

第二步：
有了上面那个函数，接下来就应该让树A的每个节点作为根节点来和B树进行比较。
遍历树A的每个节点，可用遍历算法。这里采用先序遍历。

因此，结合两个函数。可得到整个函数的代码

class Solution {
public:
    bool dfs(TreeNode* r1, TreeNode *r2){
        if(!r2)
            return true;
        if(!r1)
            return false;
        return r1->val == r2->val && dfs(r1->left, r2->left) && dfs(r1->right, r2->right);
    }
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if(!pRoot1 || !pRoot2)
            return false;
        return dfs(pRoot1, pRoot2) || HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
    }
};