题意
请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
题解—判断本层是奇数层还是偶数层
使用队列对二叉树进行层序遍历。
如果当前层是奇数层,则遍历顺序是 从左到右 ,就依次将当前层的节点值插入的 temp 临时变量的尾部(temp临时变量是存放每一层的节点值的);
如果当前层是偶数层,则遍历顺序是从右到左,就依次将当前层的节点值插入到temp临时变量的最前面,这样就可以达到 当前层最左边的节点值在最后,当前层最右边的节点值在最前面 的效果。
话不多说,上代码:
C++实现
/**
* 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>> levelOrder(TreeNode* root)
{
if(nullptr==root)
return {};
vector<vector<int>> res;
queue<TreeNode*> que;
que.push(root);
int flag = -1; //flag=1表示当前层是奇数层,flag=-1表示当前层是偶数层
while(!que.empty())
{
flag*=-1;
int size = que.size(); //本层节点数
vector<int> temp;
for(int i=0;i<size;++i)
{
TreeNode* node = que.front();
que.pop();
if(flag>0)
{
temp.push_back(node->val);
}
else if(flag<0)
{
temp.insert(temp.begin(),node->val);
}
if(node->left)
que.push(node->left);
if(node->right)
que.push(node->right);
}
//将本层遍历结果添加到结果中
res.push_back(temp);
}
//队列为空,表示已经全部遍历完毕,返回结果即可
return res;
}
};