先序遍历非递归实现
从根节点开始向左遍历,因为是先序遍历,所以每次刚遇到的节点直接输出就行,并把节点加入栈中。单次while (p != nullptr)
内循环结束,就代表以开始节点为子树的左子树已经遍历完毕,接下来就需要遍历最底层子树的右节点,即从栈中取出栈顶,也就是之前遍历左子树的最后一个节点,指向其的右儿子,如果有右儿子,则开始遍历以右儿子为根节点的子树的左子树,开始循环,如果没有右儿子,也就是p == nullptr
的情况下,则取出下一个栈顶,直到栈为空为止。
void goPreOrder(BintreeNode *& root) { stack<BintreeNode*> st; BintreeNode* p = root; while (!st.empty() || p != nullptr) { while (p != nullptr) { cout << p->data; st.push(p); p = p->child[0]; } if (!st.empty()) { p = st.top()->child[1]; st.pop(); } } return; }
中序遍历非递归实现
先遍历以开始节点为根节点的当前子树的左子树,一路遍历到底,期间把遍历到的节点加入栈中,而后每次取出栈顶,准备遍历以其节点为根节点的子树的右子树时,就把取出的栈顶节点输出,因为这个节点已经没有左儿子了,按中序遍历的规则,已然能输出。先遍历完左子树,遍历期间的节点加入栈顶,然后取出栈顶,并输出,再遍历以取出栈顶为根节点的子树的右子树,直到最后,期间按序输出的节点就是中序遍历的结果。
void goInOrder(BintreeNode*& root) { stack<BintreeNode*> st; BintreeNode* p = root; while (!st.empty() || p != nullptr) { while (p != nullptr) { st.push(p); p = p->child[0]; } if (!st.empty()) { p = st.top(); st.pop(); cout << p->data; p = p->child[1]; } } }
后序遍历非递归实现
后序遍历的非递归实现,则需要先遍历完所有的儿孙才可,那么我们只需要在之前两个非递归遍历的基础上进行修改,即把栈顶取出的节点,不进行弹出,直到其儿孙节点都遍历结束,也就是第二次从栈顶取出节点的时候,再把它输出。这样输出的序列,即是后序遍历的结果。
void goPostOrder(BintreeNode*& root) { stack<pair<BintreeNode*, bool>> st; BintreeNode* p = root; while (!st.empty() || p != nullptr) { while (p != nullptr) { st.push(make_pair(p, false)); p = p->child[0]; } if (!st.empty()) { pair<BintreeNode*, bool> &nowNode = st.top(); if (nowNode.first->child[1] == nullptr || nowNode.second) { cout << nowNode.first->data; st.pop(); } else { nowNode.second = true; p = nowNode.first->child[1]; } } } }