[PAT解题报告] Tree Traversals Again
给定二叉树非递归中序遍历出入栈的顺序,求这个二叉树的后序遍历。
关键是还原二叉树。
没必要真正还原,我们直接转变为后序遍历序列,我们也用一个堆栈模拟。
遇到一个push操作,我们把该节点放入堆栈,然后标记该节点是个新的节点,设置标签0 (0表示新, 1表示旧)
遇到一个pop操作,是该节点左孩子访问完了退回到该节点的情形,我们并不真正pop出去,我们把堆栈顶的元素设置标签1,表示现在这个节点在中序时应该已经出栈了,我们整准备访问它的右孩子节点。前提条件是:堆栈顶部的标签不是1,
如果顶部标签是1,我们应该先pop出去。
总结一下:
原始push就push同时打上标签0,表示我们正在访问这个节点的左孩子
原始pop,先从堆栈里pop出标签为1的节点,表示这些节点已经pop出来了,同时输出,因为我们的堆栈是后序遍历的序列。再把栈顶的标签设置为1,表示我们要继续访问栈顶的右孩子。
关键是对遍历顺序的理解。
代码:
#include <cstdio> #include <stack> #include <cstring> #include <string> using namespace std; stack<pair<int,int> > st; char s[10]; int main() { int n; bool mark = false; scanf("%d",&n); for (n <<= 1; n; --n) { scanf("%s",s); if (s[1] == 'u') { //push int x; scanf("%d",&x); st.push(make_pair(x,0)); } else { while ((!st.empty()) && (st.top().second == 1)) { if (mark) { putchar(' '); } else { mark = true; } printf("%d",st.top().first); st.pop(); } if (!st.empty()) { st.top().second = 1; } } } while (!st.empty()) { if (mark) { putchar(' '); } else { mark = true; } printf("%d",st.top().first); st.pop(); } puts(""); return 0; }
原题链接: http://www.patest.cn/contests/pat-a-practise/1086