后序没错,中序有错,无法检测只交换了部分的节点的BST错误。检查是否只交换部分节点的方法有错。
如果不加检测方法,只有最后一个测试点通不过。加了如下方法,则第四个测试点通不过。
#include<iostream>
using namespace std;
int ino[1002],post[1002],pre[1002],N,tag1=0,tag2=0;
void creat(int pl,int pr,int &in,int &po)
{
if(pl>=pr)return;
int next=pl+1;
if(next==pr){//叶子节点
ino[in]=pre[pl];post[po]=pre[pl];
++po;++in;
return;
}
int ll;
if(pre[next]>=pre[pl]){ //应该是左大右小,
tag1=1;
while(next<pr&&pre[next]>=pre[pl])++next;
ll=next;
while(next<pr&&pre[next]<pre[pl])++next;
if(next==ll)tag1=0;
if(tag2||next!=pr){//不是二叉搜索树
printf("NO");
exit(0);
}
creat(pl+1,ll,in,po);
ino[in]=pre[pl];++in;
creat(ll,pr,in,po);
post[po]=pre[pl];++po;
}else{ //左小右大
tag2=1;
while(next<pr&&pre[next]<pre[pl])++next;
ll=next;
while(next<pr&&pre[next]>=pre[pl])++next;
if(next==ll)tag2=0;
if(tag1||next!=pr){//不是二叉搜索树
printf("NO");
exit(0);
}
creat(pl+1,ll,in,po);
ino[in]=pre[pl];++in;
creat(ll,pr,in,po);
post[po]=pre[pl];++po;
}
}
int main()
{
freopen("test.txt","r",stdin);
scanf("%d",&N);
int i,j;
for(i=0;i<N;++i)scanf("%d",&pre[i]);
i=0,j=0;
creat(0,N,i,j);
for(i=0;i<N;++i)printf("%d ",ino[i]);
printf("\n");
for(i=0;i<N;++i)printf("%d ",post[i]);
return 0;
}
再附上一个错的方法:哈哈
#include<iostream>
#include<algorithm>
using namespace std;
int pre[1002],in[1002],post[1002],N;
void creat(int inl,int inr,int prel,int &po)
{
if(inl>inr)return ;
int i=inl;
while(i<inr&&pre[prel]!=in[i])++i;
creat(inl,i-1,prel+1,po);
creat(i+1,inr,prel+i-inl+1,po);
post[po]=pre[prel];++po;
}
int main()
{
scanf("%d",&N);
int i;
for(i=0;i<N;++i){
scanf("%d",&pre[i]);
in[i]=pre[i];
}
sort(in,in+N);
i=0;
creat(0,N-1,0,i);
if(i!=N){
printf("NO");
}else{
printf("YES\n");
for(i=0;i<N-1;++i)printf("%d ",post[i]);
printf("%d",post[N-1]);
}
return 0;
}
最后附上别人代码:大佬就是不一样。https://blog.csdn.net/lianwaiyuwusheng
#include <cstdio>
#include <vector>
using namespace std;
bool isMirror;
vector<int> pre, post;
void getpost(int root, int tail) {
if(root > tail) return ;
int i = root + 1, j = tail;
if(!isMirror) {
while(i <= tail && pre[root] > pre[i]) i++;
while(j > root && pre[root] <= pre[j]) j--;
} else {
while(i <= tail && pre[root] <= pre[i]) i++;
while(j > root && pre[root] > pre[j]) j--;
}
if(i - j != 1) return ;
getpost(root + 1, j);
getpost(i, tail);
post.push_back(pre[root]);
}
int main() {
int n;
scanf("%d", &n);
pre.resize(n);
for(int i = 0; i < n; i++)
scanf("%d", &pre[i]);
getpost(0, n - 1);
if(post.size() != n) {
isMirror = true;
post.clear();
getpost(0, n - 1);
}
if(post.size() == n) {
printf("YES\n%d", post[0]);
for(int i = 1; i < n; i++)
printf(" %d", post[i]);
} else {
printf("NO");
}
return 0;
}
其他方法基本是用建树的方法来判断。给一个博主所用的比较特别的方法https://blog.csdn.net/tjj1998/article/details/79940335
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<algorithm>
#define INF 99999999
using namespace std;
int N,pre[1010],tree[1010][2],mirror=-1;
void createTree(int s,int root){
if(pre[s]<pre[root]){
if(mirror==-1)mirror=0;
if(!tree[root][0])tree[root][0]=s;
else createTree(s,tree[root][0]);
}
if(pre[s]>=pre[root]){
if(mirror==-1)mirror=1;
if(!tree[root][1])tree[root][1]=s;
else createTree(s,tree[root][1]);
}
}
int ccount=0,flag=0;
bool Preorder(int root){
if(root!=ccount++)flag=1;
if(mirror==0){
if(tree[root][0])Preorder(tree[root][0]);
if(tree[root][1])Preorder(tree[root][1]);
}
if(mirror==1){
if(tree[root][1])Preorder(tree[root][1]);
if(tree[root][0])Preorder(tree[root][0]);
}
if(flag==1)return false;
return true;
}
int tflag=0;
void Inorder(int root){
if(mirror==0){
if(tree[root][0])Inorder(tree[root][0]);
if(tree[root][1])Inorder(tree[root][1]);
}
if(mirror==1){
if(tree[root][1])Inorder(tree[root][1]);
if(tree[root][0])Inorder(tree[root][0]);
}
if(tflag!=0)printf(" %d",pre[root]);
else tflag=1,printf("%d",pre[root]);
}
int main(){
scanf("%d",&N);
for(int i=0;i<N;i++){
scanf("%d",&pre[i]);
if(i!=0)createTree(i,0);
}
if(Preorder(0)==false)printf("NO\n");
else printf("YES\n"),Inorder(0);
return 0;
}
---------------------
作者:幻世至上
来源:CSDN
原文:https://blog.csdn.net/tjj1998/article/details/79940335?utm_source=copy
版权声明:本文为博主原创文章,转载请附上博文链接!