二叉树的遍历

来源:互联网 发布:java redis list 存取 编辑:程序博客网 时间:2024/05/19 20:01

原题链接:F - 二叉树的遍历
给出一棵二叉树的中序和前序遍历,输出它的后序遍历。
Input
本题有多组数据,输入处理到文件结束。

每组数据的第一行包括一个整数n,表示这棵二叉树一共有n个节点。

接下来的一行每行包括n个整数,表示这棵树的中序遍历。

接下来的一行每行包括n个整数,表示这棵树的前序遍历。

3<= n <= 100

Output
每组输出包括一行,表示这棵树的后序遍历。
Sample Input
7
4 2 5 1 6 3 7
1 2 4 5 3 6 7

Sample Output
4 5 2 6 7 3 1

前序遍历:当前节点->左节点->右节点
中序遍历:左节点->当前节点->右节点
后序遍历:左节点->右节点->当前节点

首先按前序遍历的顺序依次访问各节点 访问过程中 通过中序遍历 重建以当前节点为根的左子树和右子树
假设前序遍历的当前节点为c c在中序遍历的位置为m m左侧就是c的左子树 右侧就是右子树 然后同理递归
例如当前节点是 [1] 2 4 5 3 6 7 那么当前树的根是1 左右子树是 4 2 5和6 3 7
下一个节点是1 [2] 4 5 3 6 7 那么2的左右子树是4和5 依次类推
每层递归执行复杂度为0n的线性搜索 最坏的情况下复杂度为On^2

#include <iostream>#include <vector>#include <string>#include <algorithm>using namespace std;int n,pos;vector<int>pre,in,post;void rec(int l,int r){    if(l>=r)    return;    int root=pre[pos++];    int m=distance(in.begin(),find(in.begin(),in.end(),root));    rec(l,m);    rec(m+1,r);    post.push_back(root);}int main(){    int k,i;    cin>>n;    for(i=0;i<n;i++)    {        cin>>k;        pre.push_back(k);    }    for(i=0;i<n;i++)    {        cin>>k;        in.push_back(k);    }    pos=0;    rec(0,pre.size());    for(i=0;i<n;i++)    {        if(i)        cout<<" ";        cout<<post[i];    }    cout<<endl;    return 0;}
原创粉丝点击