要实现二叉树的Z字形打印,确实可以利用两个栈来分别存储奇数层和偶数层的节点,这样做可以轻松地在遍历时改变方向。这里是具体的步骤和示例代码(假设使用C++):
- 创建两个栈
stack<TreeNode*>
,一个用于存储当前层的节点(currentLevel
),另一个用于存储下一层的节点(nextLevel
)。 - 使用一个布尔值
leftToRight
来指示遍历的方向。如果是从左到右,先将左子节点压入nextLevel
栈,然后是右子节点;如果是从右到左,先将右子节点压入nextLevel
栈,然后是左子节点。 - 遍历
currentLevel
栈,对于每个节点,执行打印操作,并按leftToRight
的值将其子节点按相反顺序压入nextLevel
栈。 - 当
currentLevel
栈为空时,交换currentLevel
和nextLevel
栈,反转leftToRight
的值,以改变遍历的方向。 - 重复步骤3和4,直到两个栈都为空。
下面是C++代码示例:
#include <iostream>
#include <stack>
#include <vector>
using namespace std;
// 定义二叉树节点结构
struct TreeNode {
int val;
TreeNode *left, *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
void printZigzag(TreeNode* root) {
if (!root) return;
// 两个栈,一个用于当前层的遍历,一个用于下一层节点的存储
stack<TreeNode*> currentLevel, nextLevel;
currentLevel.push(root);
// 控制打印方向
bool leftToRight = true;
while (!currentLevel.empty()) {
TreeNode* node = currentLevel.top();
currentLevel.pop();
if (node) {
cout << node->val << " ";
// 控制遍历方向
if (leftToRight) {
if (node->left) nextLevel.push(node->left);
if (node->right) nextLevel.push(node->right);
} else {
if (node->right) nextLevel.push(node->right);
if (node->left) nextLevel.push(node->left);
}
}
// 当前层遍历完毕,准备遍历下一层
if (currentLevel.empty()) {
swap(currentLevel, nextLevel);
leftToRight = !leftToRight;
cout << endl; // 换行,表示新的一层开始
}
}
}
int main() {
// 构建和使用二叉树...
TreeNode* root = new TreeNode(1);
root->left = new TreeNode(2);
root->right = new TreeNode(3);
root->left->left = new TreeNode(4);
root->left->right = new TreeNode(5);
root->right->left = new TreeNode(6);
root->right->right = new TreeNode(7);
printZigzag(root);
return 0;
}
这段代码首先创建了一个简单的二叉树,然后使用printZigzag
函数按Z字形顺序打印出来。每次循环都打印当前层的所有节点,并按指定方向将子节点存入下一层的栈中,直到遍历完所有层级。