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

求二叉树的层序遍历

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;
   }
};

全部评论

相关推荐

10-19 00:57
门头沟学院 Java
我不是嘉心糖捏:我刚收到面试捏
投递360集团等公司6个岗位
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

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