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);
}
};