二叉树层序遍历 之字形遍历问题

简单写一下初级的层序遍历和进阶版层序遍历

    void rightSideView(TreeNode* root) {
        queue<TreeNode*> q;
        
        q.push(root);
        while (!q.empty())
        {
            TreeNode* tmp = q.front();
            cout<<"cur val is "<<tmp->val<<endl;
            q.pop();
            if(tmp->left != NULL) q.push(tmp->left);
            if(tmp->right != NULL) q.push(tmp->right);
        }

    }

    //20_1117 临时贴一个二叉树层序遍历进阶版(每一层保存在一个数组中,最终是一个二维数组)
    vector<vector<int>> getTreeNodeFloorPrint(TreeNode* root)
    {
        vector<vector<int>> res;
        if(root == NULL) return res;
        queue<TreeNode*> q;
        q.push(root);
        while (!q.empty())
        {
            int count = 0;
            int qSize = q.size();
            vector<int> tmp;
            while (count < qSize)
            {
                TreeNode* cur = q.front();
                q.pop();
                tmp.push_back(cur->val);
                if(cur->left != NULL) q.push(cur->left);
                if(cur->right != NULL) q.push(cur->right);
                count++;
            }
            res.push_back(tmp);
        }

        return res;    

    }


全部评论

相关推荐

不要停下啊:大二打开牛客,你有机会开卷了,卷起来,去找课程学习,在牛客上看看大家面试笔试都需要会什么,岗位有什么需求就去学什么,努力的人就一定会有收获,这句话从来都经得起考验,像我现在大三了啥也不会,被迫强行考研,炼狱难度开局,啥也不会,找工作没希望了,考研有丝丝机会
点赞 评论 收藏
分享
06-27 15:29
门头沟学院 Java
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务