题解 | #求二叉树的层序遍历#

求二叉树的层序遍历

https://www.nowcoder.com/practice/04a5560e43e24e9db4595865dc9c63a3


class Solution {
public:
   		vector<vector<int> > levelOrder1(TreeNode* root) {
        vector<vector<int>> res;
        if (root == nullptr)
            return res;
        queue<TreeNode*> q;
        q.push(root);
        TreeNode* cur;
        while (!q.empty()) {
            vector<int> ret;
            int n = q.size();

            for (int i = 0; i < n; ++i) {
                cur = q.front();
                q.pop();
                ret.push_back(cur->val);
                if (cur->left)
                    q.push(cur->left);
                if (cur->right)
                    q.push(cur->right);
            }
            res.push_back(ret);
        }
        return res;
    }
    void levelOrder_(TreeNode* root,int n,vector<vector<int>>& res)
    {
        if(root==nullptr)
            return;
        else
         {
            if(res.size()<n)
                res.push_back(vector<int>{});
            res[n-1].push_back(root->val);
         }
        levelOrder_(root->left, n+1, res);
        levelOrder_(root->right, n+1, res);
    }
    vector<vector<int> > levelOrder(TreeNode* root) {
        vector<vector<int>> res;
        levelOrder_(root,1,res);
        return res;
   }
};

全部评论

相关推荐

不愿透露姓名的神秘牛友
07-04 14:23
steelhead:你回的有问题,让人感觉你就是来学习的
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
昨天 12:10
点赞 评论 收藏
分享
自学java狠狠赚一...:骗你点star的,港卵公司,记得把star收回去
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
05-28 12:15
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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