JZ17 树的子结构**

题目描述

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

思路

两步走:
(1)在树A中找到和树B的根节点的值一样的节点R
(2)判断树A中以R为根节点的子树是不是包含和树B一样的结构

第一步在树A中查找与B根节点一样的节点,需要对树进行遍历(递归)
注意:一定要注意空指针
当树A或树B为空的时候,定义相应的输出。如果没有进行检查并进行相应的处理,程序会非常容易崩溃!!!
代码如下:

bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        bool result=false;
        //只有当树1和树2都不为空时才有继续的必要,否则直接返回false
        if(pRoot1!=NULL&&pRoot2!=NULL)
        {
            if(pRoot1->val==pRoot2->val)  //找到树1中节点与树2中节点相同的
            {
                result=DoesTree1HaveTree2(pRoot1,pRoot2);  //接下去判断相同节点下的这棵树的结构是否相同
            }
            //如果找不到,就继续找树1的左子树
            if(!result)
                result=HasSubtree(pRoot1->left,pRoot2);
            //如果还是找不到,就继续找树1的右子树
            if(!result)
                result=HasSubtree(pRoot1->right,pRoot2);
        }
        return result;
    }

上述代码中递归调用HasSubtree遍历二叉树A,如果发现某一节点的值和树B的头结点的值相同,则调用DoesTree1HaveTree2,进行第二步判断
第二步:判断树A中以R为根节点的子树是不是和树B具有相同的结构。同样可以用递归的思路来考虑:如果节点R的值和树B的根节点不同,则以R为根节点的子树和树B肯定不具有相同的节点;如果它们的值相同,则递归地判断它们各自的左右节点的值是否相同。递归的终止条件时到达了树A或树B 的叶子节点。
代码如下:

    bool DoesTree1HaveTree2(TreeNode* pRoot1,TreeNode*pRoot2)
    {
        //下面四行是递归的终止条件:到达了树A或树B的叶子节点
        //树2遍历结束,返回true
        if(pRoot2==NULL)
            return true;
        //树2遍历完了,树1还没有遍历完,返回false
        if(pRoot1==NULL)
            return false;
        //如果某一个节点不一样,则返回false
        if(pRoot1->val!=pRoot2->val)
        {
            return false;
        }
        //分别判断两棵树的左右节点是否相等
        return DoesTree1HaveTree2(pRoot1->left,pRoot2->left)&&DoesTree1HaveTree2(pRoot1->right, pRoot2->right);
    }

代码

class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        bool result=false;
        //只有当树1和树2都不为空时才有继续的必要,否则直接返回false
        if(pRoot1!=NULL&&pRoot2!=NULL)
        {
            if(pRoot1->val==pRoot2->val)  //找到树1中节点与树2中节点相同的
            {
                result=DoesTree1HaveTree2(pRoot1,pRoot2);  //接下去判断相同节点下的这棵树的结构是否相同
            }
            //如果找不到,就继续找树1的左子树
            if(!result)
                result=HasSubtree(pRoot1->left,pRoot2);
            //如果还是找不到,就继续找树1的右子树
            if(!result)
                result=HasSubtree(pRoot1->right,pRoot2);
        }
        return result;
    }
    //判断两棵树的结构是否一样
    bool DoesTree1HaveTree2(TreeNode* pRoot1,TreeNode*pRoot2)
    {
        //树2遍历结束,返回true
        if(pRoot2==NULL)
            return true;
        //树2遍历完了,树1还没有遍历完,返回false
        if(pRoot1==NULL)
            return false;
        //如果某一个节点不一样,则返回false
        if(pRoot1->val!=pRoot2->val)
        {
            return false;
        }
        //分别判断两棵树的左右节点是否相等
        return DoesTree1HaveTree2(pRoot1->left,pRoot2->left)&&DoesTree1HaveTree2(pRoot1->right, pRoot2->right);
    }
};
全部评论

相关推荐

叁六玖:你看,最后不是让你加油,就是鼓励你,还祝福你求职顺利。
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务