思路
题目给出一个先序序列,分析如下:
- 根据BST的性质,对于一个结点,它的左子树的所有结点的值小于该结点的值,它的右子树的所有结点的值小于等于该结点的值。
- 对于先序序列,第一个是根的值,之后是根的左子树的值的序列,最后是根的右子树的值的序列,如下图所示:
只要找出左右子树的分界,就可以递归地根据先序序列构造后序序列。可以分别从序列左侧和序列右侧开始向中间扫描,最后确定左右子树的分界线。如果得到的分界线不符合要求,直接让序列构造函数返回,于是构造后序序列成功的标志是:得到的后序序列的长度为n。 - 如果判断给出的先序序列不是BST,则将左右子树划分规则反转,再次构造后序序列,如果构造成功,则为镜像BST。
代码
#include<cstdio>
#include<algorithm>
#include<vector>
using namespace std;
vector<int> preorder;
vector<int> postorder;
int n;
bool isMirror = false;
void findPostOrder(int root, int tail) {
if (root > tail)
return;
int i = root + 1, j = tail;
if (!isMirror) {
while (i <= tail && preorder[i] < preorder[root])
i++;
while (j > root && preorder[j] >= preorder[root])
j--;
}
else {
while (i <= tail && preorder[i] > preorder[root])
i++;
while (j > root && preorder[j] <= preorder[root])
j--;
}
if (i - j != 1)
return;
findPostOrder(root + 1, j);
findPostOrder(i, tail);
postorder.push_back(preorder[root]);
}
int main() {
scanf("%d", &n);
preorder.resize(n);
for (int i = 0;i < n;i++) {
scanf("%d", &preorder[i]);
}
findPostOrder(0, n - 1);
if (postorder.size() != n) {
postorder.clear();
isMirror = true;
findPostOrder(0, n - 1);
}
if (postorder.size() == n) {
printf("YES\n");
for (int i = 0;i < postorder.size();i++) {
if (i != 0) {
printf(" ");
}
printf("%d", postorder[i]);
}
}
else {
printf("NO\n");
}
}
另解:根据给出的先序序列构造BST,接着用构造的BST的先序序列和给出的先序序列进行对比,如果两者相同说明给出的先序序列是BST。
#include<iostream>
#include<vector>
using namespace std;
struct node {
int val;
node *left, *right;
node(int v) {
val = v;
left = NULL;
right = NULL;
}
}*root = NULL;
void insert(node *&cur, int val) {
if (cur == NULL) {
cur = new node(val);
return;
}
if (cur->val > val)
insert(cur->left, val);
else insert(cur->right, val);
}
vector<int> pre;
void preorderTraverse(node *cur) {
if (cur == NULL)
return;
pre.push_back(cur->val);
preorderTraverse(cur->left);
preorderTraverse(cur->right);
}
void exchange(node *cur) {
if (cur == NULL)
return;
swap(cur->left, cur->right);
exchange(cur->left);
exchange(cur->right);
}
vector<int> res;
void postorderTraverse(node *cur) {
if (cur == NULL)
return;
postorderTraverse(cur->left);
postorderTraverse(cur->right);
res.push_back(cur->val);
}
int main() {
vector<int> preorder;
int n;
cin >> n;
preorder.resize(n);
for (int i = 0;i < n;i++) {
cin >> preorder[i];
insert(root, preorder[i]);
}
preorderTraverse(root);
bool isBST = true;
for (int i = 0;i < n;i++) {
if (preorder[i] != pre[i]) {
isBST = false;
break;
}
}
if (!isBST) {
exchange(root);
pre.clear();
preorderTraverse(root);
isBST = true;
for (int i = 0;i < n;i++) {
if (preorder[i] != pre[i]) {
isBST = false;
break;
}
}
}
if (isBST) {
cout << "YES" << endl;
postorderTraverse(root);
for (int i = 0;i < res.size();i++) {
if (i != 0)
cout << " ";
cout << res[i];
}
}
else {
cout << "NO";
}
}