题目描述:
从上到下按层打印二叉树,同一层的结点按从左到右的顺序打印,每一层打印到一行。
样例
输入如下图所示二叉树[8, 12, 2, null, null, 6, null, 4, null, null, null]
8
/ \
12 2
/
6
/
4
输出:[[8], [12, 2], [6], [4]]
算法:
使用两个辅助队列q1, q2,遍历某一层队列q1的元素时,将下一层的元素放到另一个队列q2中。然后交换两个队列的顺序,也就是说。遍历某一层队列q2的元素时,将下一层的元素放到另一个队列q1中。
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector<vector<int>> printFromTopToBottom(TreeNode* root) {
vector<vector<int> >value;
if(root==NULL)
return value;
queue<TreeNode* > q1;
queue<TreeNode* > q2;
q1.push(root);
while(!q1.empty() || !q2.empty())
{
vector<int>v1;
while(!q1.empty())
{
TreeNode* node = q1.front();
q1.pop();
v1.push_back(node->val);
if(node->left)
q2.push(node->left);
if(node->right)
q2.push(node->right);
}
if(v1.size()>0)
value.push_back(v1);
vector<int>v2;
while(!q2.empty())
{
TreeNode* node = q2.front();
q2.pop();
v2.push_back(node->val);
if(node->left)
q1.push(node->left);
if(node->right)
q1.push(node->right);
}
if(v2.size()>0)
value.push_back(v2);
}
return value;
}
};