【剑指offer】二叉树的子结构,C++实现(递归)

原创博文,转载请注明出处!

《牛客链接》

1.题目

      输入两棵二叉树A,B,判断B是不是A的子结构。(ps:约定空树不是任意一个树的子结构)

image

图1.二叉树A和二叉树B

2.思路(递归)

      查找二叉树B是否是二叉树A的子结构,可以分为两步:

第一步递归遍历树A,在树A中找到和树B的根节点的值一样的节点R。递归出口是遍历完二叉树A或在树A中找到和树B结构一样的子树。

  • 树A为空的时候,输出false
  • 树B为空的时候,输出false

第二步递归遍历树A的子树,判断树A中以R为根节点的子树是不是包含和树B一样的结构。递归出口是到达树A或者树B的叶节点。

  • 如果树B的指针为空,则返回true
  • 如果树B的指针不为空,而树A的指针为空,则返回false

image                                     

图2.树A的根节点和树B的根节点的值相同,但树A的根节点下面的结构和树B的结构不一致

 

image

图3.在树A中找到第二个值为8的节点,该节点存在和B的结构一样的子树

3.代码

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
};*/
class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        // 递归出口
        if(!pRoot1||!pRoot2)
            return false;

        return ( dfs(pRoot1,pRoot2)) || HasSubtree(pRoot1->left, pRoot2) || HasSubtree(pRoot1->right, pRoot2);
    }
private:
    bool dfs(TreeNode * r1, TreeNode * r2)
    {
        // 递归出口
        if(!r2)
            return true;
        
        // 递归出口
        if(!r1)
            return false;
        
        // 递归出口
        if(r1->val != r2->val)
            return false;
        else
            return dfs(r1->left, r2->left) && dfs(r1->right, r2->right);
    }
};

4.测试用例

# 边界测试

  • 树A的头结点是空指针
  • 树B的头结点是空指针
  • 树A和树B的头结点是空指针
  • 树A和树B只有左子节点或右子节点

# 功能测试

  • 普通二叉树,树B是树A的子结构
  • 普通二叉树,树B不是树A的子结构
posted @ 2018-03-18 10:58  wanglei5205  阅读(1031)  评论(0编辑  收藏  举报
levels of contents