数据结构作业——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

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

Sample Output

1 6 9 5 8 10 7 2 3 4
5 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;  
}