其他分享
首页 > 其他分享> > pta L2-006 树的遍历(根据后序中序确定层序)

pta L2-006 树的遍历(根据后序中序确定层序)

作者:互联网

题目链接:https://pintia.cn/problem-sets/994805046380707840/problems/994805069361299456

对树的遍历考的很周到,第一次接触到确定层序序列的题;

题目思路:建树函数和根据前中序序列确定后序序列思想是相同的,但是在细节方面还是存在差异,比如:

 

 我们可以根据题目的要求画个图

拿样例来看:

我们可以确定根节点是后序序列的最后一个数,同时我们也可以确定左右子树的位置和元素的个数还有具体的元素,那我们每次就可以确定左右子树的范围了,同时建立左右子节点的子树,就比如,

左子树的后序序列就是从起始位置到1所在的位置,也就是起始位置加上左子树的元素个数在减去1,左子树的中序序列就是从中序序列的起始位置在到根节点-1的位置,同理:

右子树的后序序列和中序序列所对应的位置也可以推导出来;

那建树的函数就解决了;

剩下的层序序列的确定就可以交给层序遍历来确定了

Talk is cheap. Show me the code.

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 int n;
 4 int in[110];//中序序列 
 5 int post[110];//后序序列 
 6 struct node
 7 {
 8     int value;
 9     node *L;
10     node *R;
11     node(int value=0,node *L=NULL,node *R=NULL):value(value),L(L),R(R){}
12 };
13 node *build_tree(int postL,int postR,int inL,int inR)//左子树的起点,终点,右子树的起点,终点 
14 {
15     if(postL>postR)//后序序列长度<=0 
16     return NULL;//返回错误 
17     node *root=new node;//新建根节点 
18     root->value=post[postR];//后序序列最后一个肯定是根节点 
19     int flag=-1;//在中序序列中查找根节点的位置 
20     for(register int i=inL;i<=inR;i++)
21     {
22         if(in[i]==post[postR])
23         {
24             flag=i;
25             break;
26         }
27     }
28     int numL=flag-inL;//左子树的个数 
29     root->L=build_tree(postL,postL+numL-1,inL,flag-1);//左子树 
30     root->R=build_tree(postL+numL,postR-1,flag+1,inR);//右子树 
31     return root;
32 }
33 int cnt;
34 void bfs(node *root)//层序遍历 
35 {
36     queue<node *>q;
37     q.push(root);
38     while(!q.empty())
39     {
40         node *newroot=q.front();//取出首元素 
41         q.pop();
42         cout<<newroot->value;//访问 
43         cnt++;
44         if(cnt<n)
45         printf(" ");
46         if(newroot->L!=NULL)
47         q.push(newroot->L);
48         if(newroot->R!=NULL)
49         q.push(newroot->R); 
50     }
51 }
52 int main()
53 {
54     scanf("%d",&n);
55     for(register int i=1;i<=n;i++)
56     scanf("%d",&post[i]);
57     for(register int i=1;i<=n;i++)
58     scanf("%d",&in[i]);
59     node *root=build_tree(1,n,1,n);
60     bfs(root);
61     return 0;
62 }

 

标签:node,左子,后序,int,中序,pta,postL,L2,序列
来源: https://www.cnblogs.com/LQS-blog/p/16174901.html