题解 | #把二叉树打印成多行#

把二叉树打印成多行

http://www.nowcoder.com/practice/445c44d982d04483b04a54f298796288

利用队列+对每一层求节点数目

  • 代码如下:
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int>> ret;
            if(pRoot==nullptr) return ret;
            queue<TreeNode*> que;
            que.push(pRoot);
            while(!que.empty()){
                int sz=que.size();//对每一层遍历求节点数目大小
                vector<int> tmp;
                while(sz){//循环一层所有节点
                    TreeNode* node = que.front();
                    que.pop();
                    tmp.push_back(node->val);
                    --sz;
                    if(node->left!=nullptr) que.push(node->left);
                    if(node->right!=nullptr) que.push(node->right);
                }
                ret.push_back(tmp);
            }
            return ret;
        }
};
  • 时间复杂度: 遍历所有的节点,因此为O(N)
  • 空间复杂度: 用了一个队列,最大使用空间为树的宽度B,因此为O(B)
全部评论

相关推荐

评论
点赞
收藏
分享

创作者周榜

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