【剑指Offer】从上往下打印二叉树

【题目】

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

【思路】

我们可以开一个辅助队列，先把根结点入队。

每次出队一个节点，并把该节点的左右子节点入队（子节点非空）。重复此操作，直至辅助队列变为空。

【代码】

/*
struct TreeNode {
	int val;
	struct TreeNode *left;
	struct TreeNode *right;
	TreeNode(int x) :
			val(x), left(nullptr), right(nullptr) {
	}
};*/
class Solution {
public:
	vector<int> PrintFromTopToBottom(TreeNode* root) {
		vector<int> result;
		if (root == nullptr) {
			return result;
		}
		queue<TreeNode*> q;  // 辅助队列
		q.push(root); 
		while (!q.empty()) {
			TreeNode *tmp = q.front();
			result.push_back(tmp->val);
			if (tmp->left != nullptr) {
				q.push(tmp->left);
			}
			if (tmp->right != nullptr) {
				q.push(tmp->right);
			}
             q.pop();
		}
		return result;
	}
};