输入两棵二叉树A和B,判断B是不是A的子结构。
分析:
- 在树A中查找与根节点值一样的结点,若找到,继续判断左右子树是否相等;若找不到,则判断以当前根节点的左右子树为根节点的子树。
- 涉及两个递归。一个是根节点遍历的递归。一个是根结点相同时,左右子树是否相同的递归。
// 判断根结点是否相同的递归
bool HasSubTree(BinaryTreeNode * pRoot1, BinaryTreeNode * pRoot2)
{
bool result = false;
if(pRoot1 != nullptr && pRoot2 != nullptr)
{
// 相同则有戏,继续判断剩下的是否相同
if(pRoot1->value == pRoot2->value)
result = DoesTree1HasTree2(pRoot1, pRoot2);
// 不同则继续看左子树
if(!result)
result = HasSubTree(pRoot1->pLeft, pRoot2);
// 还不同则继续看右子树
if(!result)
result = HasSubTree(pRoot1->pRight, pRoot2);
}
return result;
}
// 判断两棵根结点相同的树,是否左右子树相同
bool DoesTree1HasTree2(BinaryTreeNode * pRoot1, BinaryTreeNode * pRoot2)
{
// 若B已完,说明截至目前相同可返回
if(pRoot2 == nullptr)
return true;
// 若A已完但B未完(若B完就返回了),说明目前肯定不同
if(pRoot1 == nullptr)
return false;
// 如果当前结点不同,则返回
if(pRoot1->value != pRoot2->value)
return false;
// 走到此处说明要继续往下比较,仅当左、右都相同时才会返回true
return (DoesTree1HasTree2(pRoot1->pLeft, pRoot2->pLeft)
&&
DoesTree1HasTree2(pRoot1->pRight, pRoot2->pRight));
}
递归代码逻辑清楚,非常优美。