再次申明:本博客仅供个人pat复习之用,某些方法借鉴于某某博客开头会申明
本题解答来源:https://www.liuchuo.net/archives/2153
总结:
这段代码的精妙之处在于以下几个方面:
1.按常规出发:判断是否是二叉搜索树--》判断是否是镜像二叉搜索树-》是的话分别求后续遍历
否的话--》cout<<"NO";
该代码首先假定为二叉搜索树---》在假定为镜像二叉搜索树
用post.size()来判断假定是否出错-----------------------很巧妙
2.巧妙的求后续遍历的方法-----不局限于树的表达形式,值得学习。
代码:
#include<iostream>
#include<vector>
using namespace std;
vector<int> pre, post;
bool ismirr = false;
void getpost(int st,int end)
{
int root = st;
int i = root + 1; int j = end;
if (!ismirr)
{
while (i <= end&&pre[i] < pre[root])i++; //找大于root的第一个数(从左到右)
while (j>root&&pre[j] >= pre[root])j--; //找小于root的第一个数(从右到左)
}
else{ //如果是镜像二叉搜索树的话
while (i <= end&&pre[i]>=pre[root])i++; //找小于root的第一个数(从左到右)
while (j>root&&pre[j]<pre[root])j--; //找大于root的第一个数(从右到左)
}
if (i - j != 1)return;
getpost(st+1,j);
getpost(i,end);
post.push_back(pre[root]);
}
int main()
{
int n;
cin >> n;
pre.resize(n);
for (int i = 0; i < n; i++)
cin >> pre[i];
getpost(0,n-1);
if (post.size() != n)
{
ismirr = true;
post.clear();
getpost(0,n-1);
}
if (post.size() != n)
cout << "NO";
else
{
cout << "YES"<<endl;
for (int i = 0; i < n; i++)
{
cout << post[i];
if (i != post.size() - 1)cout << " ";
}
}
return 0;
}