PAT甲级1020 Tree Traversals
题目大意
本题为简单的树形数据结构问题,根据后续遍历和中序遍历,求出层次遍历。
解题思路
后续遍历的顺序,最后一个为根节点,倒数第二个为根节点的右子树的根节点,而第n-r-1位对应根节点的左子树的根节点(n为节点总数,r为右子树所有节点的总数),因此我们可以根据上述规律进行递归,构造出一颗树,然后再进行层次遍历。
层次遍历思路
用队列存储节点。刚开始让根节点入队,然后每次将队列进行出队操作时,把出队的节点的左右孩子节点加入到队列里,队列空则停止。
下附代码
#include<bits/stdc++.h>
#define N 100
using namespace std;
//树的节点结构体
struct node{
node *r=NULL;
node *l=NULL;
int id;
};
int n,a[N],b[N];
queue<node *>c;
//构造树,pos为当前对应的后续遍历的节点位置,即当前子树的根节点
//l为该子树在中序遍历中的左边界,r为右边界
node * fun(int pos,int l,int r){
node* node1=new node;
node1->id=a[pos];
if(l==r)
return node1;
if(l>r||l<0||r>n)
return NULL;
for(int i=l;i<=r;i++)
if(a[pos]==b[i]){
node1->r=fun(pos-1,i+1,r);
node1->l=fun(pos-(r-i)-1,l,i-1);
break;
}
return node1;
}
//层次遍历
void fun2(node *root){
c.push(root);
while(!c.empty()){
if(c.front()->l!=NULL)
c.push(c.front()->l);
if(c.front()->r!=NULL)
c.push(c.front()->r);
cout<<c.front()->id;
if(c.size()!=1)
cout<<" ";
else
cout<<endl;
c.pop();
}
}
int main(){
node* root;
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
cin>>b[i];
root=fun(n,1,n);
fun2(root);
return 0;
}
希望大家都可以顺利做出本题