题目描述:
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
解题思路:
举例说明:
如上图所示,因为按层打印的顺序决定应该先打印根结点,所以我们从树的根结点开始分析。为了接下来能够打印值为8的结点的两个子结点,我们应该在遍历该结点时把值为6和10的两个结点保存到一个容器里,现在容器内就有两个结点了。按照从左到右打印的要求,我们先取出为6的结点。打印出值6之后把它的值分别为5和7的两个结点放入数据容器。此时数据容器中有三个结点,值分别为10、5和7。接下来我们从数据容器中取出值为10的结点。注意到值为10的结点比值为5、7的结点先放入容器,此时又比这两个结点先取出,这就是我们通常说的先入先出,因此不难看出这个数据容器应该是一个队列。由于值为5、7、9、11的结点都没有子结点,因此只要依次打印即可。
整个打印过程如下图所示:
代码实现(c )
/*
struct treenode {
int val;
struct treenode *left;
struct treenode *right;
treenode(int x) :
val(x), left(null), right(null) {
}
};*/
class solution {
public:
vector printfromtoptobottom(treenode* root) {
treenode* fr;
if(root == null){
return result;
}
que.push(root);
while(!que.empty()){
fr = que.front();
result.push_back(fr->val);
if(fr->left != null){
que.push(fr->left);
}
if(fr->right != null){
que.push(fr->right);
}
que.pop();
}
return result;
}
private:
vector result;
queue que;
};
代码实现(python)
# -*- coding:utf-8 -*-
# class treenode:
# def __init__(self, x):
# self.val = x
# self.left = none
# self.right = none
class solution:
# 返回从上到下每个节点值列表,例:[1,2,3]
def printfromtoptobottom(self, root):
# write code here
if not root:
return []
result = []
tmp = [root]
while len(tmp):
cur = tmp.pop(0)
result.append(cur.val)
if cur.left:
tmp.append(cur.left)
if cur.right:
tmp.append(cur.right)
return result