题目要求:
输入两棵二叉树A,B,判断B是不是A的子结构。(ps:我们约定空树不是任意一个树的子结构)
解决此题需要分为两步:
1.在A树中找到与B树的根节点的值相等的结点R。
2.再判断B树是否与A树中以R为根的子树具有相同的结构。
第一步:
1.递归遍历树A,在树A中找与B树的根节点的值相等的结点R。
2.若找到进行第二步操作。
3.若没找到,则返回false。
代码实现如下:
bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
{
bool res = false;
if (pRoot1 != nullptr&&pRoot2 != nullptr)
{
if (pRoot1->val == pRoot2->val)
res = IsSubtree(pRoot1, pRoot2);//第二部操作
if (!res)//说明这个结点与B根结点值不相等。或者说明进行第二部操作后,不具有相同的结构,则继续寻找。
res = HasSubtree(pRoot1->left, pRoot2);//向左子树递归寻找
if (!res)
res = HasSubtree(pRoot1->right, pRoot2);//向右子树递归寻找
}
return res;
}
第二步:
1.递归进行判断B树是否与以R为根节点的子树具有相同的结构。
2.如果R为空,B不为空说明,R已经结束,B还有结点没匹配。说明不匹配
3.如果B为空,则说明b已经全部匹配。
4.两者值不相等,不匹配。
5.递归判断整个树。
代码如下:
bool IsSubtree(TreeNode* R, TreeNode* pRoot2)
{
if (R == nullptr&&pRoot2 != nullptr)//如果R为空,且B不为空,说明不匹配。
return false;
if (pRoot2 == nullptr)//或者只要B为空,说明匹配
return true;
if (R->val != pRoot2->val)//两者值不相等,不匹配
return false;
return IsSubtree(R->left, pRoot2->left) && IsSubtree(R->right, pRoot2->right);//递归判断左右子树。
}
可以根据下图来验证:
注意:一定要检查空指针,若没有作相应的出出力,程序很容易崩溃。
总代码见:树的子结构