题目描述
从上往下打印出二叉树的每个节点,同层节点从左至右打印。
解题思路
每一次打印一个结点的时候,如果该结点有子结点,则把该结点的子结点放到一个队列的末尾。接下来到队列的头部取出最早进入队列的结点,重复前面的打印操作,直至队列中所有的结点都被打印出来。
代码
/*
struct TreeNode {
int val;
struct TreeNode *left;
struct TreeNode *right;
TreeNode(int x) :
val(x), left(NULL), right(NULL) {
}
};*/
class Solution {
public:
vector<int> PrintFromTopToBottom(TreeNode *root) {
// 装载新的结点
vector<int> newNodeArray;
if (root == NULL){
return newNodeArray;
}
vector<TreeNode *> dequeTreeNode;
// 将根结点加入队列中
dequeTreeNode.push_back(root);
while (dequeTreeNode.size()){
TreeNode *node = dequeTreeNode.front();
vector<TreeNode *>::iterator k = dequeTreeNode.begin();
dequeTreeNode.erase(k);
newNodeArray.push_back(node->val);
if (node->left){
dequeTreeNode.push_back(node->left);
}
if (node->right){
dequeTreeNode.push_back(node->right);
}
}
return newNodeArray;
}
};