题解 | 二叉树的层序遍历

求二叉树的层序遍历

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

解题思路

题目的意思即层序遍历。 我们使用队列来实现,如果root==NULL,则直接返回空向量。否则按层扫描节点,每扫描一个节点,入队该节点的孩子节点,之后出队该节点,当队列为空时结束扫描。 使用一个计数器来判断何时分层,即当计数器的值等于当前层节点数时,该层结束。

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> ans;
        vector<int> temp;
        if(root==nullptr) return ans;
        ans.push_back(temp);
        ans[0].push_back(root->val);
        if(root->left!=nullptr) q.push(root->left);
        if(root->right!=nullptr) q.push(root->right);
        int count = 0;
        int n = q.size();
        temp.clear();
        while(n>0){
            temp.push_back(q.front()->val);
            count++;
            if(q.front()->left!=nullptr) q.push(q.front()->left);
            if(q.front()->right!=nullptr) q.push(q.front()->right);
            q.pop();
            if(count==n){
                n = q.size();
                count = 0;
                ans.push_back(temp);
                temp.clear();
            }
        }
        return ans;
    }

    queue<TreeNode*> q;
};

时间复杂度:O(N)。其中N为节点数。

空间负责度:O(L)。其中L为最长层的节点数。

全部评论

相关推荐

09-29 17:44
已编辑
蔚来_测(准入职员工)
//鲨鱼辣椒:见不了了我实习了四个月上周再投筛选了一天就给我挂了
点赞 评论 收藏
分享
1 收藏 评论
分享
牛客网
牛客企业服务