剑指offer 树的子结构

剑指offer题型分类及各题的代码及解题思路

1、题目描述

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

2、初始结构定义如下：

/*
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)
    {}
};

3、分析：

• 首先判断两个二叉树是否为空，若任意一个二叉树为空，则直接返回false。
• 递归思想，如果根节点相同则递归调用IsSubTree（），如果根节点不相同，则判断TreeA的左子树和TreeB左子树是否相同，再判断右子树和TreeB是否相同，若都相等，则返回true，否则，返回false。
• 在IsSubTree（）函数中，如果TreeB为空，则说明第二棵树遍历完了，即匹配成功，返回true。
• 若TreeA为空有两种情况：
  1. 如果treeA为空&&treeB不为空说明不匹配；
  2. 如果treeA为空，treeB为空，说明匹配。

其中需要注意的是：

• 要养成一个习惯，对任何一个树节点进行访问时，一定要提前检测该节点是否为空

4、C++实现代码如下：

class Solution 
{
public:
    //根节点已经相等，判断左右节点是否都相等，
    //若其他结点也相等，则返回true；
    //否则，返回false；
    bool IsSubTree(TreeNode* pRootA, TreeNode* pRootB)
    {
        //说明B只有一个结点，则满足B为A的子结构
        if(pRootB == NULL)
            return true;
        
        //A只有一个结点，而B不止一个结点，则说明B不可能为A的子结构，返回false
        if(pRootA == NULL)
            return false;
        
        //递归比较其他结点
        return (pRootA->val == pRootB->val) && IsSubTree(pRootA->left, pRootB->left) && IsSubTree(pRootA->right, pRootB->right);
    }
    
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        //根据题意，若两个二叉树中有一个为空，则返回false
        if(pRoot1 == NULL || pRoot2 == NULL)
            return false;
        //否则，判断两个二叉树根节点值是否相等，
        //若根节点值相等，则继续判断找左右节点值是否相等；
        //若根节点值不相等，则分别遍历两个二叉树，继续判断
        return IsSubTree(pRoot1, pRoot2) || HasSubtree(pRoot1->left, pRoot2)
            || HasSubtree(pRoot1->right, pRoot2);
    }
};