数据结构 - 二叉树的构建 (分冶)
L2-2. 树的遍历
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
陈越
给定一棵二叉树的后序遍历和中序遍历,请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。
输入格式:
输入第一行给出一个正整数N(<=30),是二叉树中结点的个数。第二行给出其后序遍历序列。第三行给出其中序遍历序列。数字间以空格分隔。
输出格式:
在一行中输出该树的层序遍历的序列。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:7 2 3 1 5 7 6 4 1 2 3 4 5 6 7输出样例:
4 1 6 3 5 7 2
直接看代码吧,利用递归构建二叉树,利用队列输出。
#include <iostream> #include <string> #include <sstream> #include <queue> #define MAX 35 using namespace std; int In[MAX],Out[MAX],ltr[MAX],rtr[MAX]; int n; int build_tr(int l1,int r1,int l2,int r2) { if(l1 > r1) return 0; int root = Out[r2]; int cur=l1; while(In[cur] != root) cur++; int cnt = cur-l1; ltr[root] = build_tr(l1,cur-1,l2,l2+cnt-1); //??????????? rtr[root] = build_tr(cur+1,r1,l2+cnt,r2-1); //??????????? return root; } void Out_put() { queue<int > a; int temp; int root=Out[n-1]; int flag=0; a.push(root); while(!a.empty()){ temp=a.front(); if(!flag){ cout<<temp; flag=1; } else cout<<" "<<temp; a.pop(); if(ltr[temp]) a.push(ltr[temp]); if(rtr[temp]) a.push(rtr[temp]); } cout<<endl; } int main() { while(cin>>n) { for(int i = 0;i < n;i ++) cin>>Out[i]; for(int i = 0;i < n;i ++) cin>>In[i]; build_tr(0,n-1,0,n-1); Out_put(); } return 0; }