本文最后更新于:2020年6月28日 下午

题目描述:

从上往下打印出二叉树的每个节点,同层节点从左至右打印。

实现如下:

//层次遍历时使用队列容器,保证先进先出
//分别判断节点的左右孩子是否为空
//非空就将孩子节点尾插
//每次都从队头获取打印值、头删出队
//队列为空时说明遍历结束
//注意判断root是否合法
class Solution 
{
public:
	deque<TreeNode*> q;//存放层次遍历时的节点顺序
	vector<int> vec;

	vector<int> PrintFromTopToBottom(TreeNode* root) 
	{
		if (root == NULL) return vec;//root为空时,直接return
		q.push_back(root);//root非空时,首先尾插
		while (!q.empty())//判断是否遍历结束
		{
			TreeNode *tmp = q.front();//队头获取该打印的节点
			q.pop_front();
			//分别判断左右孩子是否为空,非空时队列尾插
			if (tmp->left != NULL)
				q.push_back(tmp->left);
			if (tmp->right != NULL)
				q.push_back(tmp->right);
			vec.push_back(tmp->val);//将val添加
		}
		return vec;
	}
};