题目:请实现一个函数按照之字形顺序打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右到左的顺序打印,第三行再按照从左到右的顺序打印,其他行以此类推。
思路:按之字形打印二叉树需要两个栈,在打印某一层的节点时,把下一层的子节点保存到相应的栈里。
1.如果当前打印的是奇数层,则先保存左子节点再保存右子节点;如果当前打印的是偶数层,则先保存右子节点再保存左子节点
2.在打印一个栈里的节点时,它的子节点保存到另一个栈里
3.当一层所有节点都打印完毕时,交换这两个栈并继续打印下一层
核心代码如下:
struct BinaryTreeNode{
int m_nValue;
BinaryTreeNode* m_pLeft;
BinaryTreeNode* m_pRight;
};
void Print(BinaryTreeNode* pRoot){
if(pRoot == nullptr)
return;
stack<BinaryTreeNode*> levels[2]; //创建一个栈数组,包含 levels[0]和 levels[1]
int current = 0;
int next = 1;
levels[current].push(pRoot);
while(!levels[0].empty() || !levels[1].empty()){
BinaryTreeNode* pNode = levels[current].top();
levels[current].pop();
cout << pNode->m_nValue;
if(current == 0){
//先保存左节点,后保存右节点
if(pNode->m_pLeft != nullptr)
levels[next].push(pNode->m_pLeft);
if(pNode->m_pRight != nullptr)
levels[next].push(pNode->m_pRight);
}
else{
//先保存右节点,后保存左节点
if(pNode->m_pRight != nullptr)
levels[next].push(pNode->m_pRight);
if(pNode->m_pLeft != nullptr)
levels[next].push(pNode->m_pLeft);
}
if(levels[current].empty()){
cout << endl;
current = 1 - current; //交换两个栈
next = 1 - next;
}
}
}