您的位置:首页 > 职场人生

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

2017-07-26 19:44 537 查看
思路整理自剑指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;
}
};
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  剑指Offer