题目描述:给定一个二叉树,返回该二叉树的之字形层序遍历,(第一层从左向右,下一层从右向左,一直这样交替)
示例:
该二叉树之字形层序遍历的结果是 [ [1], [3,2], [4,5] ]
解题思路:我们可以先了解一下二叉树的层序遍历,可以采用数据结构的堆来实现,可利用容器queue。首先将根结点加入队首,之后依次执行将队首的左右子节点加入队尾,并将队首pop弹出,直到队列为空。
之字形的层序遍历与上述方法相似,此处的想法是可以借助两个栈结构来实现。两个栈分别储存奇数层和偶数层的结点。首先将根节点将入储存奇数层结点的栈,在每一层遍历结束的时候,如果是奇数层,则储存偶数层结点的栈应该为空;如果是偶数层,则储存奇数层结点的栈应该为空。如果偶数层节点的栈为空,则应取出储存奇数层结点的栈的首个元素,将其按右节点、左节点的顺序加入储存偶数层的栈,依次循环,直至存储奇数层的栈为空。如果储存奇数层结点的栈为空,则应取出偶数层结点的栈的首个元素,按期左节点、右节点的顺序加入储存奇数层结点的栈,直至储存偶数节点的栈为空。如果两个站都为空,则结束循环,即可完成之字形的层序遍历。
代码实现如下:、
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int>> ans;
stack<TreeNode*>odd;
stack<TreeNode*>even;
even.push(pRoot);
vector<int> an;
if(pRoot==nullptr)
return ans;
while(odd.size()||even.size())
{
if(even.empty())
{
while(!odd.empty())
{
TreeNode* pTreeNode = odd.top();
an.push_back(pTreeNode->val);
odd.pop();
if(pTreeNode->right)
even.push(pTreeNode->right);
if(pTreeNode->left)
even.push(pTreeNode->left);
}
ans.push_back(an);
an.clear();
}
else if(odd.empty())
{
while(!even.empty())
{
TreeNode* pTreeNode = even.top();
an.push_back(pTreeNode->val);
even.pop();
if(pTreeNode->left)
odd.push(pTreeNode->left);
if(pTreeNode->right)
odd.push(pTreeNode->right);
}
ans.push_back(an);
an.clear();
}
}
return ans;
}