剑指Offer-26：树的子结构

题目：

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

链接：

剑指Offer（第2版）：P148

思路标签：

• 数据结构：树
• 算法：递归

解答：

1. C++

• 第一步遍历树A，找到与树B根结点一样的节点R；找到则进行第二部分，否则进行递归查找；
• 第二步判断树A中以R为根结点的子树是不是和树B具有相同的子结构；使用递归依次判断当前节点和左右子树；
• 注意边界退出的条件和可能会报错的情况；
• 注意判断两个double值是否相等的方法，不能直接使用“==”.

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};*/

class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        bool result = false;

        if(pRoot1 != nullptr && pRoot2 != nullptr){
            if(Equal(pRoot1->val, pRoot2->val))
                result = DoesTree1HaveTree2(pRoot1, pRoot2);
            if(!result)
                result = HasSubtree(pRoot1->left, pRoot2);
            if(!result)
                result = HasSubtree(pRoot1->right, pRoot2);
        }

        return result;
    }

    bool DoesTree1HaveTree2(TreeNode* pRoot1, TreeNode* pRoot2){
        // 找到结束条件
        if(pRoot2 == nullptr)
            return true;

        // 未找到异常情况
        if(pRoot1 == nullptr)
            return false;

        if(!Equal(pRoot1->val, pRoot2->val))
            return false;

        return DoesTree1HaveTree2(pRoot1->left, pRoot2->left) && DoesTree1HaveTree2(pRoot1->right, pRoot2->right);
    }

    // double 值相等的处理
    bool Equal(double num1, double num2){
        return (num1 - num2 > -0.0000001) && (num1 - num2 < 0.0000001);
    }
};