设二叉树采用二叉链表表示,指针root指向根结点,试编写一个在二叉树中查找值为x的结点,并打印该结点所有祖先结点的算法。在此算法中,假设值为x的结点不多于一个。
【解答】
此算法采用后序的非递归遍历形式。因退栈时需要区分其左、右子树是否已经遍历,
故在结点进栈时附带有一个标志,=0,进入左子树,=1,进入右子树。
用栈S1保存结点指针ptr,
用栈S2保存标志tag。
这两个栈的进栈与退栈是同步的。
#include <iostream.h>
#include “stack.h”
template <class Type>
void BinaryTree <Type> ::Find - Print(BinTree <Type> & BT; Type x) {
Stack < BinTreeNode <Type>* > S1; Stack <int> S2;
BinTreeNode <Type>* t = BT.root; int i, top;
while (t != NULL && t->data != x || !S1.IsEmpty()) {
while (t != NULL && t->data != x) {//寻找值为x的结点
S1.Push(t); S2.Push(0); //进栈, 作向左子树遍历标志
t = t->leftChild; //转向左子树
}
if (t != NULL && t->data == x) { //找到值为x的结点
while (!S1.IsEmpty()) {
cout << S1.GetTop()->data << “ << “;
S1.Pop();
}
break;
}
else { //未找到值为x的结点
while (!S2.IsEmpty() && S2.GetTop() == 1) {
S1.Pop(); S2.Pop();
}
if (!S2.IsEmpty()) {
S2.Pop(); S2.Push(1); //改向右子树遍历的标志
t = S1.GetTop()->rightChild; //转向右子树
}
}
}
}