数据结构作业——order（二叉树遍历）

order

Description

给出一棵二叉树的中序遍历和每个节点的父节点，求这棵二叉树的先序和后 序遍历。

Input

输入第一行为一个正整数 n 表示二叉树的节点数目， 节点编号从 1 到 n，其中 1 为根节点。
第 2 行有 n 个数字， 第 i 个数字表示 i 的父亲节点。（ 1 的父亲节点为 0， 表示无）
第 3 行为中序遍历。
30%的数据： n<=20；
60%的数据： n<=1000；
100%的数据： n<=10000；

Output

输出 2 行， 第一行为先序遍历，第二行为后序遍历。

Sample Input

100 7 2 2 9 1 8 1 6 89 5 6 1 10 8 7 3 2 4

Sample Output

1 6 9 5 8 10 7 2 3 45 9 6 10 3 4 2 7 8 1

思路

存下每个节点的左右儿子，然后根据中序遍历判断谁是左儿子，谁是右儿子，建完树，跑一下先序遍历和后序遍历

 

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn = 10005;
int lson[maxn],rson[maxn],pos[maxn];
bool first = true;  

void PreOrder(int x)
{
    if (x != 1) printf(" ");
    printf("%d",x);
    if (lson[x])    PreOrder(lson[x]);
    if (rson[x])    PreOrder(rson[x]);
}  

void PostOrder(int x)
{
    if (lson[x])    PostOrder(lson[x]);
    if (rson[x])    PostOrder(rson[x]);
    first?printf("%d",x):printf(" %d",x);
    first = false;
}  

int main()
{
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    int N,i,tmp;
    memset(lson,0,sizeof(lson));
    memset(rson,0,sizeof(rson));
    memset(pos,0,sizeof(pos));
    scanf("%d",&N);
    for (i = 1;i <= N;i++)
    {
        scanf("%d",&tmp);
        if (tmp == 0)   continue;
        if (!lson[tmp]) lson[tmp] = i;
        else    rson[tmp] = i;
    }
    for (i = 1;i <= N;i++)
    {
        scanf("%d",&tmp);
        pos[tmp] = i;
    }
    for (i = 1;i <= N;i++)
    {
        if (lson[i] && rson[i])
        {
            if (pos[rson[i]] < pos[lson[i]])
            {
                swap(rson[i],lson[i]);
            }
        }
        else if (!lson[i] && rson[i])
        {
            if (pos[rson[i]] < pos[i])
            {
                lson[i] = rson[i];
                rson[i] = 0;
            }
        }
        else if (lson[i] && !rson[i])
        {
            if (pos[lson[i]] > pos[i])
            {
                rson[i] = lson[i];
                lson[i] = 0;
            }
        }
    }
    PreOrder(1);printf("\n");
    PostOrder(1);printf("\n");
    return 0;
}