【剑指Offer】面试题18:树的子结构

来源:互联网 发布:spss数据分析师认证 编辑:程序博客网 时间:2024/06/06 01:44

思路整理自剑指Offer

一:题目描述

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


二:解题思路

要查找树A中找到和树B结构一样的子树:

第一步:在树A中找到和B的根节点的值一样的结点R

第二步:判断树A中以R为根节点的子树是不是包含树B一样的结构


以一个例子进行说明:

1.我们在树A中找到值为8的结点:从A的根节点开始遍历,我们发现它的根节点的值就是8

2.接着我们就去判断树A的根节点下面的子树是否含有和树B一样的结构。

3.在树A中,根节点的左子树的值是8,而树B的根节点的左子树结点是9,对应的两个结点不同。

4.因此我们仍然要遍历树A,接着查找的值为8的结点,我们在树的第二层找到一个值为8的结点

5.判断该结点下面的子树是否含有和B树一样结构的子树,先后找到两个结点9,2,这和树B的结构完全相同。因此树B是树A的子结构

以一个例子说明:

三:代码实现

/*struct TreeNode {int val;struct TreeNode *left;struct TreeNode *right;TreeNode(int x) :val(x), left(NULL), right(NULL) {}};*/class Solution {public:        bool DoesTree1HaveTree2(TreeNode* pRoot1, TreeNode* pRoot2){       //树B遍历完,则说明树B是以pRoot为根节点的子树,是A的子树        if(pRoot2==NULL)            return true;        //如果B还没有遍历结束,pRoot1就已经遍历完,则B不是子树        if(pRoot1==NULL)            return false;                //如果对应结点不相等,则B不是子树        if(pRoot1->val!=pRoot2->val)            return false;                //如果当前对应结点符合条件,则分别遍历左子结点和右子节点        return DoesTree1HaveTree2(pRoot1->left,pRoot2->left) && DoesTree1HaveTree2(pRoot1->right,pRoot2->right);    }        bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)    {bool result=false;                //在每一处需要访问地址的时候都要问自己这个地址有没有可能是NULL,如果是NULL,如何处理        if(pRoot1==NULL || pRoot2==NULL)            return result;                //如果A的R节点与B的根节点相同,那么我们要判断A中以R结点为根节点的子树是否包含和树B一样的结构        if(pRoot1->val==pRoot2->val)            result=DoesTree1HaveTree2(pRoot1,pRoot2);        //如果R为根节点的子树不包含B,则继续遍历A,寻找与B树根节点相同的结点        //左子结点        if(!result)            result=HasSubtree(pRoot1->left,pRoot2);        //右子节点        if(!result)            result=HasSubtree(pRoot1->right,pRoot2);                return result;    }};

原创粉丝点击