寻找树的子结构,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

欢迎大家交流指正!!!

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务