题意:给定一串键值,判断该序列是否是一颗二叉搜索树的先序遍历或者是镜像的先序遍历,若是打印该该数的后序遍历,
思路:不用重构二叉树,直接根据二叉搜索树的性质完成,先判断是否是原树的先序遍历,若是直接打印结果然后在判断是否是其镜像树的先序遍历,最后打印结果。
二叉搜索树的先序的性质:先序遍历的顺序为:根左子树右子树,如果是先序遍历,则左子树都小于根,右子树全部大于等于根
代码:
#include<iostream>
#include<vector>
using namespace std;
vector<int> pre, post;
int n = 0;
bool isMirro = false;//首先假设不是镜像的二叉树
void getPost(int root,int end) {
if (root > end) return;
int i = root + 1, j = end;
if (!isMirro) {
while (i <= end && pre[root] > pre[i])i++;
while (j > root && pre[root] <= pre[j])j--;
}
else {
while (i <= end && pre[root] <= pre[i])i++;
while (j > root && pre[root] > pre[j])j--;
}
if (i - j != 1)return;//不是先序遍历就退出
getPost(root + 1, j);
getPost(i, end);
post.push_back(pre[root]);
}
int main() {
cin >> n;
pre.resize(n);
for (int i = 0; i < n; i++) scanf("%d",&pre[i]);
getPost(0, n - 1);
if (post.size() != n) {//不是原数的先序遍历
isMirro = true;
post.clear();
getPost(0, n - 1);
}
if (post.size() == n) {
printf("YES\n");
for (int i = 0; i < n; i++) {
printf("%d%s", post[i], i == n - 1 ? "\n" : " ");
}
}
else {
printf("NO\n");
}
system("pause");
return 0;
}