JZ26 树的子结构

输入两棵二叉树A,B,判断B是不是A的子结构。(我们约定空树不是任意一个树的子结构)

假如给定A为{8,8,7,9,2,#,#,#,#,4,7},B为{8,9,2},2个树的结构如下,可以看出B是A的子结构

代码:

#define FALSE 0
#define TRUE  1
bool has_sub(struct TreeNode* pRoot1, struct TreeNode* pRoot2){
    if(pRoot1 == NULL && pRoot2 != NULL){
        return FALSE;
    }
    if(pRoot2 == NULL){
        return TRUE;
    }
    if(pRoot1->val != pRoot2->val){
        return FALSE;
    }
    return has_sub(pRoot1->left, pRoot2->left) && has_sub(pRoot1->right, pRoot2->right);
}

bool pre_order(struct TreeNode* pRoot1, struct TreeNode* pRoot2) {
    if(pRoot1 == NULL){
        return FALSE;
    }
    if(pRoot1->val == pRoot2->val){
        if(has_sub(pRoot1, pRoot2)){
            return TRUE;
        }
    }
    return pre_order(pRoot1->left, pRoot2) || pre_order(pRoot1->right, pRoot2);
}
bool HasSubtree(struct TreeNode* pRoot1, struct TreeNode* pRoot2 ) {
    if(pRoot1 == NULL || pRoot2 == NULL){
        return FALSE;
    }
    return pre_order(pRoot1, pRoot2);
}

一个复杂的递归问题。

首先要思考,如何判断是不是子树。首先根节点值相同,而且后续的每个节点都要相同。

我们对主树进行前序遍历(根左右),首先判断根节点是不是等于子树的根节点,如果是,那接着判断剩余节点;如果不等于,那就判断主树的子节点是不是等于子树的根节点。它们之间的关系为或,即只要找到一个证明是子树,那就可以返回true。

对于判断其余节点是不是相同同样适用前序遍历,当主树为NULL,而子树有值时,说明不是子树;当值不相等时也为false,当所有之前的值都相等,而且子树为NULL时,则说明是子树。要求所有节点都相等所以是与的逻辑关系。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值