DS搜索树判断

题目描述

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

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

输入

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

输出

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

输入样例1

7
8 6 5 7 10 8 11

输出样例1

YES
5 7 6 8 11 10 8

NOTICE：

根据输入创建二叉顺序树,然后将先序遍历结果与输入序列比较,相同则再输出后序序列即可

#include <iostream>
using namespace std;

int* a;//存储输入
int* b;//二叉树前序遍历结果
int* c;//二叉树后序遍历结果
int bindex = 0;
int cindex = 0;

class BiTreeNode
{
private:
	int data;
	BiTreeNode* lchild;
	BiTreeNode* rchild;
	BiTreeNode* parent;
public:
	BiTreeNode(int a, BiTreeNode* par)
	{
		data = a;
		lchild = nullptr;
		rchild = nullptr;
		parent = par;
	}
	friend class BiTree;
};

class BiTree
{
private:
	BiTreeNode* root;

	void PreOrder(BiTreeNode* t)
	{
		if (t)
		{
			b[bindex++] = t->data;
			PreOrder(t->lchild);
			PreOrder(t->rchild);
		}
	}
	void LastOrder(BiTreeNode* t)
	{
		if (t)
		{
			LastOrder(t->lchild);
			LastOrder(t->rchild);
			c[cindex++] = t->data;
		}
	}
public:
	BiTree() :root(nullptr) {}
	void insert(int num)
	{
		if (root == nullptr)
		{
			BiTreeNode* t = new BiTreeNode(num, nullptr);
			root = t;
		}
		else
		{
			BiTreeNode* p = root;
			BiTreeNode* q = nullptr;//指向p节点的双亲
			while (1)
			{
				if (p == nullptr)
				{
					if (num < q->data)
					{
						BiTreeNode* t = new BiTreeNode(num, q);
						q->lchild = t;
					}
					else if (num >= q->data)
					{
						BiTreeNode* t = new BiTreeNode(num, q);
						q->rchild = t;
					}
					break;
				}


				if (num < p->data)
				{
					q = p;
					p = p->lchild;
				}
				else if (num >= p->data)
				{
					q = p;
					p = p->rchild;
				}
			}
		}
	}
	void PreOrder()
	{
		PreOrder(root);
	}
	void LastOrder()
	{
		LastOrder(root);
	}
};

int main()
{
	int n;
	BiTree tree;

	cin >> n;
	a = new int[n];
	b = new int[n];
	c = new int[n];
	for (int i = 0; i < n; i++)
	{
		int num;
		cin >> num;
		a[i] = num;
		tree.insert(num);
	}
	tree.PreOrder();
	tree.LastOrder();


	int flag = 0;
	for (int i = 0; i < n; i++)
	{
		if (a[i] != b[i])
		{
			flag = 1;
			break;
		}
	}
	if (flag == 1)
	{
		cout << "NO" << endl;
	}
	else
	{
		cout << "YES" << endl;
		for (int i = 0; i < n; i++)
		{
			cout << c[i];
			if (i == n - 1)
				cout << endl;
			else
				cout << " ";
		}
	}

	return 0;
}