关于树的遍历一题的思路+代码（树的后序与中序遍历输出层序遍历）（递归）

最新推荐文章于 2023-03-08 21:35:59 发布

原创最新推荐文章于 2023-03-08 21:35:59 发布 · 364 阅读

0 ·

CC 4.0 BY-SA版权

图与树专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种根据二叉树的后序遍历和中序遍历来构造树的方法，并利用广度优先搜索（BFS）输出层序遍历。首先，找到后序遍历中在中序遍历中最靠后的节点作为根节点，然后递归地构造左右子树。最后，通过BFS遍历输出层序结果。代码中用到了队列来实现BFS，但注意代码的时间复杂度较高，为O(n^3)，有待优化。

题目来源：Acwing 1497.树的遍历||PTA L2-006 树的遍历 (25 分)
21.04.19补充：回头看这个方法太麻烦了，简单些的可以参考 PTA天梯练习赛刷题笔记1 的 L2-011 玩转二叉树

题目描述

一个二叉树，树中每个节点的权值互不相同。
现在给出它的后序遍历和中序遍历，请你输出它的层序遍历。

输入描述

第一行包含整数 N，表示二叉树的节点数。
第二行包含 N 个整数，表示二叉树的后序遍历。
第三行包含 N 个整数，表示二叉树的中序遍历。

输出描述

输出一行 N 个整数，表示二叉树的层序遍历。

数据范围

1≤N≤30

输入样例1

7
2 3 1 5 7 6 4
1 2 3 4 5 6 7

输出样例1

4 1 6 3 5 7 2

思路

对于任何一个二叉树的合法中序遍历与后序遍历，中序遍历中在后序遍历中位置最靠后的数，即为该树的根；

依照此根，我们又可以将该中序遍历分为左右两部分，即为该根的左右两棵子树，进行递归处理。

建树完成之后，通过BFS获取层序遍历。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

vector<int> son[31];//son[x]代表x的子节点

int n,i,j;
int mid[31],lst[31],rot=0;//rot存总根

void maketr(int l,int r,int fa)//fa存该树的根的父节点
{
    if(l==r)//叶子
    {
        son[fa].push_back(mid[l]);
        return;
    }
    else if(l>r)return;//非法
    int root=0,rm=0;//root存该子树的根，rm存该根在中序遍历中的位置
    for(i=l;i<=r;i++)//对于该子树的每一个节点
    {
        for(j=1;j<=n;j++)//在后序遍历中位置最靠后的一个
        {
            if(mid[i]==lst[j])
            {
                if(j>root){root=j,rm=i;}
                //printf("*");
                break;
            }
        }
    }
    if(!rot)rot=lst[root];//如果总根未确定，该根即为总根
    //printf("%d\n",root);
    son[fa].push_back(lst[root]);//将此树的根连接至其父节点
    maketr(l,rm-1,lst[root]);//递归左子树
    maketr(rm+1,r,lst[root]);//递归右子树
    return;
}
queue<int> que;//bfs队列
void bfs(int x)
{
    //printf("*");
	int pf=0;
    for(i=0;i<son[x].size();i++)que.push(son[x][i]);
    while(!que.empty())
    {
        int now=que.front();
        que.pop();
		if(!pf)pf=1;
		else printf(" ");
        printf("%d",now);
        for(i=0;i<son[now].size();i++)que.push(son[now][i]);//树无自环、重边与回路，不需要考虑重复加入的问题
    }
}
int main()
{
    cin>>n;
    for(i=1;i<=n;i++)scanf("%d",&lst[i]);
    for(i=1;i<=n;i++)scanf("%d",&mid[i]);
    maketr(1,n,0);//建树
    bfs(0);//bfs输出
    return 0;
}