L2-006. 树的遍历 图论

L2-006. 树的遍历

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

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

输入格式：

输入第一行给出一个正整数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 <bits/stdc++.h>
using namespace std;
int n;
int zx[55],hx[55];

struct node
{
    int val;
    node *lc,*rc;
};
node *root;

node *build(int zl,int zr,int hl,int hr)
{
    node *rt=new node;
    rt->val=hx[hr];
    rt->lc=NULL;
    rt->rc=NULL;
    for (int i=0;zl+i<=zr;i++){
        if (zx[zr-i]==hx[hr]){
            if (zl+i<zr)
                rt->lc=build(zl,zr-i-1,hl,hr-i-1);
            if (i>0)
                rt->rc=build(zr-i+1,zr,hr-i,hr-1);
            break;
        }
    }

    return rt;
}

void levelorder(node *rt)
{
    queue<node*> q;
    q.push(rt);
    printf("%d",rt->val);
    while (!q.empty()){
        node *cur=q.front();
        q.pop();
        if (cur->lc!=NULL){
            printf(" %d",cur->lc->val);
            q.push(cur->lc);
        }
        if (cur->rc!=NULL){
            printf(" %d",cur->rc->val);
            q.push(cur->rc);
        }
    }
}

int main()
{
    scanf("%d",&n);
    for (int i=0;i<n;i++)
        scanf("%d",&hx[i]);
    for (int i=0;i<n;i++)
        scanf("%d",&zx[i]);
    root=build(0,n-1,0,n-1);
    levelorder(root);
    printf("\n");
    return 0;
}