团体天梯 L2-004 （结构与算法 7-28） 这是二叉搜索树吗？ （25 分)（50行精简）

L2-004 这是二叉搜索树吗？ （25 分)

一棵二叉搜索树可被递归地定义为具有下列性质的二叉树：对于任一结点，

• 其左子树中所有结点的键值小于该结点的键值；
• 其右子树中所有结点的键值大于等于该结点的键值；
• 其左右子树都是二叉搜索树。

所谓二叉搜索树的“镜像”，即将所有结点的左右子树对换位置后所得到的树。

给定一个整数键值序列，现请你编写程序，判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。

输入格式：

输入的第一行给出正整数 N（≤1000）。随后一行给出 N 个整数键值，其间以空格分隔。

输出格式：

如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果，则首先在一行中输出 YES ，然后在下一行输出该树后序遍历的结果。数字间有 1 个空格，一行的首尾不得有多余空格。若答案是否，则输出 NO。

输入样例 1：

7
8 6 5 7 10 8 11

输出样例 1：

YES
5 7 6 8 11 10 8

输入样例 2：

7
8 10 11 8 6 7 5

输出样例 2：

YES
11 8 10 7 5 6 8

输入样例 3：

7
8 6 8 5 10 9 11

输出样例 3：

NO

思路：建立线索二叉树，进行先序遍历，检查是否与原序列相同，相同则输出后序序列

 

#include<iostream>
#include<vector>
using namespace std;
struct Node {
	int data = 0;
	Node *left = NULL, *right = NULL;
};
vector<int> preOrder, judge, out;
void insert(Node *head, int tmp) {   //构建搜素树
	Node *r;
	if (head != NULL) {
		r = tmp < head->data ? head->left : head->right;
		if (r != NULL)	insert(r, tmp);
		else
			(tmp < head->data ? head->left : head->right) = new Node({ tmp });
	}
}
void Order(Node *head, const int& key) {
	if (head != NULL) {
		judge.push_back(head->data);
		if (!key)		Order(head->left, key);		
		Order(head->right, key);
		if (key)		Order(head->left, key);		//镜像判断
		out.push_back(head->data);  //存储后序
	}
}
int main() {
	int n, tmp;
	cin >> n >> tmp;
	Node *head = new Node({ tmp });
	preOrder.assign({ tmp });
	for (int i = 1; i < n; i++) {
		cin >> tmp;
		preOrder.push_back(tmp);
		insert(head, tmp);
	}
	for (int i = 0; i < 2; i++) {
		judge.clear();
		out.clear();
		Order(head, i);
		if (judge == preOrder) {
			cout << "YES" << endl;
			for (int j = 0; j < out.size(); j++)
				cout << (j ? " " : "") << out[j];
			return 0;
		}
	}
	cout << "NO" << endl;
	return 0;
}