PTA 7-18树的遍历

原创已于 2022-12-23 23:23:11 修改 · 230 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #c语言

于 2022-12-23 23:08:14 首次发布

该博客主要介绍了如何根据一棵二叉树的后序遍历和中序遍历序列，重建并输出其层次遍历的序列。通过输入的后序和中序遍历序列，利用算法匹配找到每个节点，并按层次顺序存储。最后，输出层次遍历的结果。

给定一棵二叉树的后序遍历和中序遍历，输出其层序遍历的序列

输入格式：

输入第一行给出一个正整数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<stdio.h>
int main(){
    //其中二维数组其中一个用于记录是否成为节结点
    //a[1],b[1]的数0，代表未成为结点，1表示已经成为结点
    int n,a[2][30],b[2][30];
    //初始化
    scanf("%d",&n);
    for(int i=0;i<n;i++){scanf("%d",&a[0][i]);a[1][i]=0;}
    for(int i=0;i<n;i++){scanf("%d",&b[0][i]);b[1][i]=0;}
    int c[30];//用于存储层次遍历的序列
    int num=0;//num记录已经成为结点的个数;
    while(num!=n){/*都成为结点的话就退出循环，每次的while（）可以视为一个层次*/
        for(int i=0;i<n;i++){
            //记录左或者右子树下的结点数
            int cnt=0;
            while(b[1][i]==0 && i<n){cnt++;i++;}//如果已经是前面结点就跳过
            //如果cnt=0,说明这个数已经成为结点要跳过不再录入
            if(cnt!=0){
                //由后序遍历可得根结点要从后序遍历的数组的后端找起
// / *这里就是单纯的匹配
                for(int j=n-1;j>=0;j--){//修改为for(int j=0;j<n;j++){便是先序与中序确定层次
                    if(a[1][j])continue;
                    int flag=0;//flag用于是否找到根结点退出循环
                    for(int k=i-cnt;k<i;k++){
                        if(a[0][j]==b[0][k]){//说明该为根结点
                            c[num++]=a[0][j];//存储结点同时num+1
                            a[1][j]=1;//结点标志
                            b[1][k]=1;//结点标志
                            flag=1;
                            break;
                        }
                    }
                    if(flag)break;
                }
//   * /
            }
        }
    }
    //打印层次遍历结果
    for(int i=0;i<n;i++){
        if(i)printf(" %d",c[i]);
        else printf("%d",c[i]);
    }
    return 0;
}