題目描述:
從上到下按層打印二叉樹,同一層的結點按從左到右的順序打印,每一層打印到一行。
樣例
輸入如下圖所示二叉樹[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;
}
};