【22】从上到下打印二叉树《剑指offer》
题目描述:从上往下打印出二叉树的每个节点,同层节点从左至右打印。
解题思路:根据题目描述,打印顺序就是层序遍历的顺序。第一次接触到该题时,想着尝试用递归的方法去解决,但是不清楚如何用递归实现层序遍历,所以用了比较笨的方法,先将树的节点依次加入到一个buffer缓存区中,然后在依次将数据打印。代码如下:
vector<int> PrintFromTopToBottom(TreeNode* root) {
vector<TreeNode*> buffer;
vector<int> res;
if(root!=NULL)
buffer.push_back(root);
int beg= 0;
unsigned int buffer_size = buffer.size();
while(beg!=buffer.size()){ //当buffer不再增加时退出
buffer_size = buffer.size();
for(auto i = beg; i<buffer_size;++i){ //这里可能比较难以理解,画图较好理解
if(buffer[i]->left!=NULL)
buffer.push_back(buffer[i]->left);
if(buffer[i]->right!=NULL)
buffer.push_back(buffer[i]->right);
}
beg = buffer_size;
}
for(auto i=buffer.begin();i!=buffer.end();++i){
cout<<(*i)->val<<"\t";
res.push_back((*i)->val);
}
return res;
}
高赞解法:
链接:https://www.nowcoder.com/questionTerminal/7fe2212963db4790b57431d9ed259701?f=discussion
来源:牛客网
public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) {
ArrayList<Integer> list = new ArrayList<>();
ArrayList<TreeNode> queue = new ArrayList<>();
if (root == null) {
return list;
}
queue.add(root);
while (queue.size() != 0) {
TreeNode temp = queue.remove(0);
if (temp.left != null){
queue.add(temp.left);
}
if (temp.right != null) {
queue.add(temp.right);
}
list.add(temp.val);
}
return list;
}
其实思路是类似的 但是他这样做,相比自己的解法 少用了依次循环,且通过queue结构的利用少用了一些内存空间,因为每一次新的node来的时候,queue已经将上一层的node释放了。
自己暴露的问题:没能善用queue(), 对于内存空间的利用率上考虑不足。