PAT1020 (已知中序,后序遍历转前序遍历)

已知后序与中序输出前序(先序):
后序:3, 4, 2, 6, 5, 1(左右根)
中序:3, 2, 4, 1, 6, 5(左根右)

已知一棵二叉树,输出前,中,后时我们采用递归的方式。同样也应该利用递归的思想:

对于后序来说,最后一个节点肯定为根。在中序中可以找到左子树的个数,那么就可以在后序中找到左子树的根;同理也可以找到右子树。

void pre(int root, int start, int end) {
    if(start > end) return ;
    int i = start;
    while(i < end && in[i] != post[root]) i++;
    printf("%d ", post[root]);
    pre(root - 1 - end + i, start, i - 1);
    pre(root - 1, i + 1, end);
}

如果想要按照层次输出数,那么我们可以增加一个index变量,来存储结点。根节点为0,左子树为index*2+1, 右子树为index*2+2,那么我们按顺序输出不是-1的节点即可

#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
vector<int>lever(100000,-1),post,in;
int n;
void pre(int root,int start,int end,int index)
{
    if(start>end)
        return;
    int i=start;
    while(i<end&&in[i]!=post[root])
        i++;
    lever[index]=post[root];
    pre(root-end+i-1,start,i-1,index*2+1);
    pre(root-1,i+1,end,index*2+2);
}
int main()
{
    scanf("%d",&n);
    post.resize(n+1);
    in.resize(n+1);
    for(int i=1;i<=n;i++)
        scanf("%d",&post[i]);
    for(int i=1;i<+n;i++)
        scanf("%d",&in[i]);
    pre(n,1,n,0);
    int cnt=0;
    for(int i=0;i<lever.size();i++)
    {
        if(lever[i]!=-1)
        {
            cnt++;
            if(cnt==n)
            {
                printf("%d\n",lever[i]);
                break;
            }
            else
                printf("%d ",lever[i]);
        }
    }
}

 

posted @ 2018-03-09 17:27  blueattack  阅读(428)  评论(0编辑  收藏  举报