1086. Tree Traversals Again (25)
An inorder binary tree traversal can be implemented in a non-recursive way with a stack. For example, suppose that when a 6-node binary tree (with the keys numbered from 1 to 6) is traversed, the stack operations are: push(1); push(2); push(3); pop(); pop(); push(4); pop(); pop(); push(5); push(6); pop(); pop(). Then a unique binary tree (shown in Figure 1) can be generated from this sequence of operations. Your task is to give the postorder traversal sequence of this tree.
Figure 1
Input Specification:
Each input file contains one test case. For each case, the first line contains a positive integer N (<=30) which is the total number of nodes in a tree (and hence the nodes are numbered from 1 to N). Then 2N lines follow, each describes a stack operation in the format: "Push X" where X is the index of the node being pushed onto the stack; or "Pop" meaning to pop one node from the stack.
Output Specification:
For each test case, print the postorder traversal sequence of the corresponding tree in one line. A solution is guaranteed to exist. All the numbers must be separated by exactly one space, and there must be no extra space at the end of the line.
Sample Input:6 Push 1 Push 2 Push 3 Pop Pop Push 4 Pop Pop Push 5 Push 6 Pop PopSample Output:
3 4 2 6 5 1
根据push得到前序序列,根据push和pop得到中序序列,然后求一下后序序列。
#include<iostream> #include<vector> #include<string> using namespace std; vector<int>pre; vector<int>in; vector<int>post; vector<int>tmp; void getpost(int s1,int e1,int s2,int e2) { if(s1>e1) return ; int mid; for(int i=s2;i<=e2;++i) if(in[i]==pre[s1]) mid=i; getpost(s1+1,mid+s1-s2,s2,mid-1); getpost(e1-e2+mid+1,e1,mid+1,e2); post.push_back(pre[s1]); } int main() { int n; cin>>n; for(int i=0;i<2*n;++i) { string op; cin>>op; if(op.size()==3) { int num=tmp[tmp.size()-1]; tmp.pop_back(); in.push_back(num); } else { int num; cin>>num; pre.push_back(num); tmp.push_back(num); } } getpost(0,n-1,0,n-1); cout<<post[0]; for(int i=1;i<post.size();++i) cout<<" "<<post[i]; return 0; }