PTA数据结构与算法题目集(中文) 7-28

PTA数据结构与算法题目集(中文)  7-28

7-28 搜索树判断 (25 分)

 

对于二叉搜索树，我们规定任一结点的左子树仅包含严格小于该结点的键值，而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树，得到的树叫做镜像二叉搜索树。

现在我们给出一个整数键值序列，请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列，如果是，则输出对应二叉树的后序遍历序列。

输入格式:

输入的第一行包含一个正整数N（≤1000），第二行包含N个整数，为给出的整数键值序列，数字间以空格分隔。

输出格式:

输出的第一行首先给出判断结果，如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列，则输出YES，否侧输出NO。如果判断结果是YES，下一行输出对应二叉树的后序遍历序列。数字间以空格分隔，但行尾不能有多余的空格。

输入样例1:

7
8 6 5 7 10 8 11

输出样例1:

YES
5 7 6 8 11 10 8

输入样例2:

7
8 6 8 5 10 9 11

输出样例2:

NO
题目分析:一道利用树的基础题 将数据读入后 再按先序遍历输出 比较 注意因为有镜像二叉搜索树 所以先序遍历要检查从左边开始以及从右边开始 最后输出后序遍历时 二叉搜索树从左边 镜像二叉搜索从右边

#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<string.h>
#include<malloc.h>

typedef struct TreeNode* PtrToTreeNode;
typedef PtrToTreeNode BTree;
struct TreeNode
{
    int Data;
    PtrToTreeNode Lc;
    PtrToTreeNode Rc;
};

PtrToTreeNode Insert(int Data, BTree BT)
{
    if (!BT)
    {
        BT = (BTree)malloc(sizeof(struct TreeNode));
        BT->Data = Data;
        BT->Lc = NULL;
        BT->Rc = NULL;
    }
    else
    {
        if (Data < BT->Data)
            BT->Lc = Insert(Data, BT->Lc);
        else
            BT->Rc = Insert(Data, BT->Rc);
    }
    return BT;
}

int Pre[1000];
int Size;
void PreOrder(BTree Bt)
{
    if (Bt)
    {
        Pre[Size++] = Bt->Data;
        PreOrder(Bt->Lc);
        PreOrder(Bt->Rc);
    }
}
void PreOrderRight(BTree Bt)
{
    if (Bt)
    {
        Pre[Size++] = Bt->Data;
        PreOrderRight(Bt->Rc);
        PreOrderRight(Bt->Lc);
    }
}

int Last[1000];
int Size2;
void LastOrder(BTree Bt)
{
    if (Bt)
    {
        LastOrder(Bt->Lc);
        LastOrder(Bt->Rc);
        Last[Size2++] = Bt->Data;
    }
}
void LastOrderRight(BTree Bt)
{
    if (Bt)
    {
        LastOrderRight(Bt->Rc);
        LastOrderRight(Bt->Lc);
        Last[Size2++] = Bt->Data;
    }
}

int main()
{
    int N;
    scanf("%d", &N);
    BTree Bt = NULL;
    int Num[1000] = { 0 };
    for (int i = 0; i < N; i++)
    {
        scanf("%d", &Num[i]);
        Bt = Insert(Num[i], Bt);
    }
    PreOrder(Bt);
    int flag = 1;
    int Left = 1;
    for (int i = 0; i < N; i++)
    {
        if (Num[i] != Pre[i])
        {
            flag = 0;
            break;
        }
    }
    if (!flag)
    {
        Size = 0;
        flag = 1;
        Left = 0;
        PreOrderRight(Bt);
        for (int i = 0; i < N; i++)
        {
            if (Num[i] != Pre[i])
            {
                flag = 0;
                break;
            }
        }
    }
    if (!flag)
        printf("NO
");
    else
    {
        printf("YES
");
        if (Left)
        {
            LastOrder(Bt);
            for (int i = 0; i < N - 1; i++)
                printf("%d ", Last[i]);
        }
        else
        {
            LastOrderRight(Bt);
            for (int i = 0; i < N - 1; i++)
                printf("%d ", Last[i]);
        }
        printf("%d", Last[N - 1]);
    }
    return 0;
}