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 h[35];
int m[35];
int n;
struct node
{
    int data;
    node* lchild;
    node* rchild;

};
vector<int> ans;
node* build(int hl,int hr,int ml,int mr)
{
    if(hl>hr||ml>mr) return NULL;
    int root1=h[hr];
    node* root=new node;
    root->data=root1;
    int record;
    for(int i=ml;i<=mr;i++)
    {
        if(m[i]==root1)
        {
            record=i;
            break;
        }
    }
    int record1=record-ml;
    root->lchild=build(hl,hl+record1-1,ml,record-1);
    root->rchild=build(hl+record1,hr-1,record+1,mr);

    return root;
}
void bfs(node* w)
{
    queue<node*> q;
    q.push(w);
    while(!q.empty())
    {
        node* now=q.front();
        q.pop();
        ans.push_back(now->data);
        if(now->lchild!=NULL) q.push(now->lchild);
        if(now->rchild!=NULL) q.push(now->rchild);
    }
}
int main()
{
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>h[i];
    }
    for(int i=0;i<n;i++)
    {
        cin>>m[i];
    }
    bfs(build(0,n-1,0,n-1));
    for(int i=0;i<ans.size();i++)
    {
        cout<<ans[i];
        if(i!=ans.size()-1) cout<<' ';
    }

    return 0;
}

---