题目
(摘自王道考研数据结构5.3.3习题第五题)
在二叉树中有两个节点m和n,若m是n的祖先节点,则使用后序遍历可以找到m到n的路径。
分析
首先需要理解的是,前中后序遍历都是通过递归的方式,这里我们以非递归的代码来理解,将后来需要用到的结点保存在栈中,比如下面这颗树:
前序遍历:根左右,过程是:根节点m入栈并输出,访问m的左孩子a,a入栈并输出,访问a的左孩子c,c入栈并输出,c没有左孩子,无元素入栈,c没有右孩子,无元素入栈,c出栈,此时栈顶元素为a,访问a元素的右孩子d,d入栈并输出,d没有左孩子,无元素入栈,d没有右孩子,无元素入栈,d出栈,a左右子树都访问完了所以出栈,现在栈顶元素是m,m已经没有作用了所以出栈,访问m的右孩子b,m的右孩子b入栈并输出,接着访问b的左孩子e,e入栈并输出……
中序遍历:左根右,过程是:根节点m入栈,访问m的左孩子,所以m的左孩子a入栈,访问a的左孩子,所以a的左孩子c入栈,c没有左孩子,c现在输出并出栈,c没有右孩子所以没有元素入栈,现在栈顶元素是a,a输出并出栈,a的右孩子d入栈,此时d没有左孩子所以没有元素入栈,d输出并出栈,d没有右孩子所以没有元素入栈,现在栈顶元素是m,m输出,此时m已经没有作用了所以出栈,m的右孩子b入栈,访问b的左孩子,所以b的左孩子e入栈……
后序遍历:左右根,过程是:根节点m入栈,访问m的左孩子,所以m的左孩子a入栈,访问a的左孩子,所以a的左孩子入栈,c没有左孩子和右孩子,没有元素入栈,c现在输出并出栈。现在栈顶元素是a,a的右孩子d入栈,d没有左孩子和右孩子,没有元素入栈,d现在输出并出栈。此刻栈顶是a,且a的左孩子右孩子均已经访问过,所以此刻a输出并出栈。此刻栈顶元素是m,m的右孩子b入栈,访问b的左孩子,所以m的左孩子b入栈,b有左孩子e,所以e入栈,e有左孩子g,所以g入栈,g没有左孩子和右孩子,没有元素入栈,g此刻输出并出栈。此刻栈顶元素是e,e有右孩子n,即目标元素之一,所以n入栈,因为n没有左孩子和右孩子,所以输出n并出栈。此刻栈顶元素是e,且e的左右孩子均已经被访问,所以输出e并出栈。此刻栈顶是b,b有孩子f,所以f入栈,f没有左右孩子,所以输出f并出栈,此刻栈顶是b,且b的左右孩子均已被访问,所以输出b并出栈。此刻栈顶元素是m,m的左右孩子均已访问,所以m出栈。
所以在前序和中序的过程中,如果n在m的右子树部分,遍历过程中找到了n,但是m已经不在栈中,因为栈中只会保留等会需要用到的e和b结点,而m已经完成了访问根结点和m的左子树的任务,已经被出栈,所以无法追溯n如何走到m。
但是后序遍历就不一样啦,后序的顺序是左右根,所以只要m的左右子树还没遍历完成,m就不能出栈,在遍历m的左右子树过程中,无论在m的左边还是右边找到了n,都可以直接返回然后根据栈中的路径让n找到回到m的路,这样就能找到m到n的路径~
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109
| typedef struct { BTNode *data[MaxSize]; int top; } SqStack;
void PreOrder1(BTNode *b) { BTNode *p; SqStack *st; InitStack(st); Push(st,b); while (!StackEmpty(st)) { Pop(st,p); printf("%c ",p->data); if (p->rchild!=NULL) Push(st,p->rchild); if (p->lchild!=NULL) Push(st,p->lchild); } printf("\n"); DestroyStack(st); } void PreOrder2(BTNode *b) { BTNode *p; SqStack *st; InitStack(st); p=b; while (!StackEmpty(st) || p!=NULL) { while (p!=NULL) { printf("%c ",p->data); Push(st,p); p=p->lchild; } if (!StackEmpty(st)) { Pop(st,p); p=p->rchild; } } printf("\n"); DestroyStack(st); } void InOrder1(BTNode *b) { BTNode *p; SqStack *st; InitStack(st); if (b!=NULL) { p=b; while (!StackEmpty(st) || p!=NULL) { while (p!=NULL) { Push(st,p); p=p->lchild; } if (!StackEmpty(st)) { Pop(st,p); printf("%c ",p->data); p=p->rchild; } } printf("\n"); } DestroyStack(st); }
void PostOrder1(BTNode *b) { BTNode *p,*r; bool flag; SqStack *st; InitStack(st); p=b; do { while (p!=NULL) { Push(st,p); p=p->lchild; } r=NULL; flag=true; while (!StackEmpty(st) && flag) { GetTop(st,p); if (p->rchild==r) { printf("%c ",p->data); Pop(st,p); r=p; } else { p=p->rchild; flag=false; } } } while (!StackEmpty(st)); printf("\n"); DestroyStack(st); }
|
补充
Q:可是问题是这个方法会夹杂着不属于路径的节点,比如n的兄弟节点?
A:如果n是左叶子结点,则会先访问n,如果n是右叶子结点有左兄弟结点,左兄弟结点被访问后会提前出栈,然后才会访问n,所以访问到n的时候,栈中剩余的元素就可以找到从n到m的路径,中间无其他结点。
参考