题解 | #把二叉树打印成多行#
把二叉树打印成多行
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)
查看13道真题和解析