二叉树的非递归遍历
- 二叉树的三种遍历方式也可以通过非递归的方法借助栈来实现。
- 通过控制节点的出栈和入栈先后顺序来实现对树的不同方式的遍历。
非递归先根遍历二叉树
- 当栈不为空或者当前节点为不为空,执行操作:
- 从根节点开始,依次访问树中最左端的节点并入栈,当节点为空停止入栈。
- 取栈顶元素为当前节点并出栈,如果当前节点有右子树,则遍历其右子树。
void PreorderPrint(SearchBTree T) // 先序根(序)遍历二叉树并打印出结点值
{
if (T == nullptr)
{
cout << "Empty tree!";
exit(1);
}
Stack S = new StackNode; // 借助栈来实现
initStack(S);
TreeNode* pT = T; // 创建临时结点指向根节点
while (pT || !isEmpty(S)) // 当结点不为空或者栈不为空执行循环
{
if (pT) // 当pT不为空
{
Push(S, pT); // pT入栈
cout << pT->data << " "; // 打印当前结点
pT = pT->left; // pT移动指向其左孩子
}
else
{ // pT为空,则
pT = getTop(S);
Pop(S); // 若pT为空表示左子树的左孩子全部遍历完,依次出栈
pT = pT->right; // 当左孩子及根结点遍历完之后,开始遍历其右子树
}
}
cout << endl;
delete S;
}
非递归中根遍历二叉树
- 当栈不为空或者当前节点为不为空,执行操作:
- 同样地,先依次将根节点及其左子树最左端的节点入栈,但不进行访问。当节点为空,则停止入栈。
- 访问栈顶元素作为当前节点并出栈,如果当前节点有右子树,则遍历访问其右子树。
void InorderPrint(SearchBTree T) // 中根(序)遍历二叉树并打印出结点值
{
if (T == nullptr)
{
cout << "Empty tree!";
exit(1);
}
Stack S = new StackNode; // 借助栈来实现
initStack(S);
TreeNode* pT = T; // 创建临时结点指向根节点
while (pT || !isEmpty(S)) // 当结点不为空或者栈不为空执行循环
{
if (pT) // 当pT不为空
{
Push(S, pT); // pT入栈
pT = pT->left; // pT移动指向其左孩子
}
else
{ // pT为空,则
pT = getTop(S);
Pop(S); cout << pT->data << " "; // 若pT为空表示左子树的左孩子全部遍历完,依次出栈并打印
pT = pT->right; // 当左孩子及根结点遍历完之后,开始遍历其右子树
}
}
cout << endl;