PAT A 1119 Pre- and Post-order Traversals (30分)

最新推荐文章于 2022-02-26 19:51:19 发布

Siumai

最新推荐文章于 2022-02-26 19:51:19 发布

阅读量185

点赞数 1

分类专栏： PAT A 文章标签： PAT PAT A

本文链接：https://blog.csdn.net/qq_43749739/article/details/104329910

版权

PAT A 专栏收录该内容

140 篇文章 1 订阅

订阅专栏

一、题目概述

判断给定先序、后序遍历序列是否可以唯一确定一个二叉树，并输出所确定的二叉树的中序遍历序列。二叉树不唯一时，构造可确定的任一二叉树中序序列。

二、思路

判断是否可以唯一确定：二叉树先序序列的第二个元素是左子树的根元素（若有左子树），后序序列的倒数第二个元素是右子树的根元素（若有右子树）。当先序第二、后序倒数第二个元素相同时可知，该元素即可为左子树的根元素，也可为右子树的根元素，二叉树不唯一。

注意：中序序列后输出换行，否则会出现测试点格式错误，应该是为了便于多个正确答案的模式匹配。

三、代码

#include <cstdio>
#include <vector>
using namespace std;
struct node
{
    int data;
    node *lchild, *rchild;
};
vector<int> pre(30), post(30);
int N, flag = 1, index = 0;
void create(node *&t, int l1, int r1, int l2, int r2)
{
    t = new(node);
    t->data = pre[l1];
    t->lchild = t->rchild = NULL;
    if(l1 == r1)
        return;
    int offset;
    for(offset = 0; post[l2 + offset] != pre[l1 + 1]; ++offset);
    create(t->lchild, l1 + 1, l1 + offset + 1, l2, l2 + offset);
    if(l2 + offset + 1 != r2)
        create(t->rchild, l1 + offset + 2, r1, l2 + offset + 1, r2 - 1);
    else flag = 0;
}
void inorder(node *t)
{
    if(t)
    {
        inorder(t->lchild);
        printf("%s%d", index++ ? " ":"", t->data);
        inorder(t->rchild);
    }
}
int main()
{
    node *root;
    scanf("%d", &N);
    for(int i = 0; i < N; ++i)
        scanf("%d", &pre[i]);
    for(int i = 0; i < N; ++i)
        scanf("%d", &post[i]);
    create(root, 0, N - 1, 0, N - 1);
    printf("%s\n", flag ? "Yes":"No");
    inorder(root);
    printf("\n");
}