寻找树的子结构,C++

class Solution {
public:
    bool HasSubtree(TreeNode* pRoot1, TreeNode* pRoot2)
    {
        if (pRoot2 == nullptr)
            return false;

        queue<TreeNode*> tqu;
        tqu.push(pRoot1);
        while(!tqu.empty()){
            int length = tqu.size();
            for (int i = 0; i != length; ++i){
                if (tqu.front() != nullptr){
                    tqu.push(tqu.front()->left);
                    tqu.push(tqu.front()->right);

                    if (pRoot2->val == tqu.front()->val){
                        // 执行判断子结构函数 若不是继续遍历pRoot1
                         if (IsSubtree(tqu.front(), pRoot2))
                             return true;
                    }
                }
                tqu.pop();
            }
        }
        return false;
    }

    bool IsSubtree(TreeNode* pRoot1, TreeNode* pRoot2){
        if (pRoot2 == nullptr)
            return true;
        else if (pRoot1 == nullptr || pRoot1->val != pRoot2->val)
            return false;
        else{
            return IsSubtree(pRoot1->left, pRoot2->left) && IsSubtree(pRoot1->right, pRoot2->right);
        }
    }

};

1、首先考虑题目提示,B为空树则不是子结构
2、接着层序遍历A,寻找A中与B头节点相同的节点
(1)找到后执行函数IsSubtree()递归函数,若判断B是子结构return true;判断不是则退出继续层序遍历
PS:题中没说A树中节点值均不相同,所以更深层可能有子结构
(2)层序完都没子结构则return false

欢迎大家交流指正!!!

全部评论

相关推荐

10-09 00:50
已编辑
长江大学 算法工程师
不期而遇的夏天:1.同学你面试评价不错,概率很大,请耐心等待;2.你的排名比较靠前,不要担心,耐心等待;3.问题不大,正在审批,不要着急签其他公司,等等我们!4.预计9月中下旬,安心过节;5.下周会有结果,请耐心等待下;6.可能国庆节前后,一有结果我马上通知你;7.预计10月中旬,再坚持一下;8.正在走流程,就这两天了;9.同学,结果我也不知道,你如果查到了也告诉我一声;10.同学你出线不明朗,建议签其他公司保底!11.同学你找了哪些公司,我也在找工作。
点赞 评论 收藏
分享
11-07 13:31
怀化学院 Java
勇敢牛牛不怕难:又疯一个
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务