【GPLT 二阶题目集】L2-006 树的遍历

给定一棵二叉树的后序遍历和中序遍历，请你输出其层序遍历的序列。这里假设键值都是互不相等的正整数。

输入格式：

输入第一行给出一个正整数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 <iostream>
#include <stdlib.h>
#include <queue>
using namespace std;

int n, inOrder[30], postOrder[30]; //中序遍历序列，后序遍历序列

typedef struct biTreeNode { //定义二叉树结点
    int key;
    biTreeNode* lchild;
    biTreeNode* rchild;
}biTree, * BiTree;

void creatBiTree(BiTree& T, int pl, int pr, int il, int ir)//建立二叉树
{
    T = (biTreeNode*)malloc(sizeof(biTreeNode));
    T->key = postOrder[pr]; //当前结点的值等于后序遍历序列的最后一个值
    int k;
    for (int i = 0; i < n; i++) {
        if (inOrder[i] == T->key) {
            k = i;
        }
    }
    if (k > il) creatBiTree(T->lchild, pl, k - il + pl - 1, il, k - 1);
    else T->lchild = NULL;
    if (k < ir) creatBiTree(T->rchild, pr - ir + k, pr - 1, k + 1, ir);
    else T->rchild = NULL;
    return;
}

void FloorPrint(BiTree& T) //对二叉树层序遍历
{
    queue<BiTree> temp;//储存二叉树的结点地址
    temp.push(T); //根结点入队
    BiTree temp2;
    while (!temp.empty()) {
        cout << temp.front()->key; //输出队头的值
        temp2 = temp.front(); temp.pop(); //队头出队
        if (!temp2->lchild && !temp2->rchild && temp.empty())
            cout << endl;
        else
            cout << " ";
        if (temp2->lchild) //左子树入队
            temp.push(temp2->lchild);
        if (temp2->rchild) //右子树入队
            temp.push(temp2->rchild);
    }
}

int main()
{
    cin >> n; 
    for (int i = 0; i < n; i++) { //获取后序遍历序列
        cin >> postOrder[i];
    }
    for (int i = 0; i < n; i++) { //获取中序遍历序列
        cin >> inOrder[i];
    }
    BiTree head; //创建头指针，指向二叉树的根结点
    creatBiTree(head, 0, n - 1, 0, n - 1); //根据后序遍历序列和中序遍历序列建立二叉树
    FloorPrint(head);
    return 0;
}

 注意事项：

这题如果把给定序列中的数字作为字符处理，部分测试点会错......

如有问题，欢迎提出。