剑指offer-18:判断树的子结构

输入两棵二叉树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));
}

递归代码逻辑清楚,非常优美。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值