【树的遍历】

分数 20

全屏浏览题目

切换布局

作者 陈越

单位 浙江大学

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数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

代码长度限制

16 KB

时间限制

400 ms

内存限制

64 MB

---

定义结点：

typedef struct TreeNode *BiTree;
struct TreeNode{
    int Data;
    BiTree Left;
    BiTree Right;
};

通过中后序创建一棵树：

BiTree CreateBiTree(int root, int start, int end) {
//root为后序根节点，start为中序开始，end为中序结束
	BiTree T;
	int i;
	if (start>end) return NULL;//中序数组个数为0 
	for (i =start; Post[root]!=In[i]; i++);//在中序数组中找根节点
	T = (TreeNode *)malloc(sizeof(struct TreeNode));//建立新结点申请空间
    T->Data=Post[root];//建立新的数的根节点为后序的根节点
	T->Left = CreateBiTree(root-(end-i)-1, start, i-1);//递归左子树
	T->Right = CreateBiTree(root-1,i+1,end);//递归右子树
    //上两步见下图
	return T;//将树返回
}

通过先序遍历进行输出：（中序为左中右 后序为左右中）

void PreOrder(BiTree T){
    if(T==NULL) return;
    else{
        cout<<" "<<T->Data;//输出数
        PreOrder(T->Left);//递归左子树
        PreOrder(T->Right);//递归右子树
    }

}

层次遍历输出：

void LevelOrder(BiTree T)
{
    queue<TreeNode> q;//建立一个队列
    int flag=1;
    if(T!=NULL)//如果树不为空
    {
        TreeNode temp;
        q.push(*T);//将根节点压入队列
        while(!q.empty())//当队列不为空时
        {
            temp=q.front();//弹出队首元素
            q.pop();//在队列中删除
            if(flag)  //输出这一层的结点
            {cout<<temp.data;
             flag=0;}
            else
            cout<<" "<<temp.data;
            if(temp.Left!=NULL) q.push(*temp.Left);//下一层所有结点入队
            if(temp.Right!=NULL) q.push(*temp.Right);
        }
    }
}

main函数：

int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)//输入后序数组
        cin>>Post[i];
    for(int i=0;i<n;i++)输入中序数组
        cin>>In[i];
    BiTree T=NULL;//定义一棵空树
    T=CreateBiTree(n-1,0,n-1); //建立一棵中序后序遍历的树
    LevelOrder(T);//层序输出
}

---

本题代码：

#include<bits/stdc++.h>
using namespace std;
const int MAX=31;
int In[MAX],Post[MAX];
//结点
typedef struct TreeNode* BiTree;
struct TreeNode{
    int data;
    BiTree Left;
    BiTree Right;
};

//中后序遍历创建一棵树
BiTree CreateBiTree(int root,int start,int end){
    BiTree T;
    int i;
    if(start>end) return NULL;
    for(i=start;Post[root]!=In[i];i++);
    T=(TreeNode *)malloc(sizeof(struct TreeNode));
    T->data=Post[root];
    T->Left=CreateBiTree(root-(end-i)-1,start,i-1);
    T->Right=CreateBiTree(root-1,i+1,end);

    return T;
}
//层次遍历输出
void LevelOrder(BiTree T)
{
    queue<TreeNode> q;//建立一个队列
    int flag=1;
    if(T!=NULL)//如果树不为空
    {
        TreeNode temp;
        q.push(*T);//将根节点压入队列
        while(!q.empty())//当队列不为空时
        {
            temp=q.front();//弹出队首元素
            q.pop();//在队列中删除
            if(flag)  //输出这一层的结点
            {cout<<temp.data;
             flag=0;}
            else
            cout<<" "<<temp.data;
            if(temp.Left!=NULL) q.push(*temp.Left);//下一层所有结点入队
            if(temp.Right!=NULL) q.push(*temp.Right);
        }
    }
}

int main()
{
    int n;
    cin>>n;
    for(int i=0;i<n;i++)//输入后序数组
        cin>>Post[i];
    for(int i=0;i<n;i++)输入中序数组
        cin>>In[i];
    BiTree T=NULL;//定义一棵空树
    T=CreateBiTree(n-1,0,n-1); //建立一棵中序后序遍历的树
    LevelOrder(T);//层序输出
}