对于二叉搜索树,我们规定任一结点的左子树仅包含严格小于该结点的键值,而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树,得到的树叫做镜像二叉搜索树。
现在我们给出一个整数键值序列,请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,如果是,则输出对应二叉树的后序遍历序列。
输入格式:
输入的第一行包含一个正整数N(≤1000),第二行包含N个整数,为给出的整数键值序列,数字间以空格分隔。
输出格式:
输出的第一行首先给出判断结果,如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列,则输出YES
,否侧输出NO
。如果判断结果是YES
,下一行输出对应二叉树的后序遍历序列。数字间以空格分隔,但行尾不能有多余的空格。
输入样例1:
7
8 6 5 7 10 8 11
输出样例1:
YES
5 7 6 8 11 10 8
输入样例2:
7
8 6 8 5 10 9 11
输出样例2:
NO
题解:先建立起二叉搜索树,然后将先序遍历和镜像先序遍历的结果分别放到数组里,然后判断是否有一个与原数组相同,只要有1个相同就输出YES,然后后序遍历即可。(看完思路应该就大开了吧~~)
#include<bits/stdc++.h>
using namespace std;
struct node
{
node *l,*r;
int data;
};
int a[10001],b[10001],c[10001];
int top,tot,flag1=1,flag2=1,flag3,flag4;
node *create(node *root,int x)
{
if(root==NULL)
{
root=new node;
root->data=x;
root->l=root->r=NULL;
return root;
}
else
{
if(x<root->data)
root->l=create(root->l,x);
else
root->r=create(root->r,x);
}
return root;
}
void show1(node *root)
{
if(root)
{
b[++top]=root->data;
show1(root->l);
show1(root->r);
}
}
void show2(node *root)
{
if(root)
{
c[++tot]=root->data;
show2(root->r);
show2(root->l);
}
}
void finallshow1(node *root)
{
if(root)
{
finallshow1(root->l);
finallshow1(root->r);
if(flag3==0)
{
cout<<root->data;
flag3=1;
}
else
cout<<" "<<root->data;
}
}
void finallshow2(node *root)
{
if(root)
{
finallshow2(root->r);
finallshow2(root->l);
if(flag4==0)
{
cout<<root->data;
flag4=1;
}
else
cout<<" "<<root->data;
}
}
int main()
{
int n;
cin>>n;
node *root=NULL;
for(int i=1;i<=n;i++)
{
cin>>a[i];
root=create(root,a[i]);//先建立二叉搜索树
}
show1(root);//分别将先序遍历和镜像先序遍历放入b和c数组里
show2(root);
for(int i=1;i<=n;i++)//只要有1个不同,就说明不是该树的遍历
{
if(a[i]!=b[i])
{
flag1=0;
break;
}
}
for(int i=1;i<=n;i++)
{
if(a[i]!=c[i])
{
flag2=0;
break;
}
}
if(flag1)
{
cout<<"YES"<<endl;
finallshow1(root);//树的遍历
}
else if(flag2)
{
cout<<"YES"<<endl;
finallshow2(root);//树的镜像遍历
}
else
cout<<"NO"<<endl;
return 0;
}