题目链接:
题目分析:
这是一道二叉查找树的问题。一般题目都是需要建树,下面给出建树的通用算法:
void insert(node* &root,int w) //建二叉查找树,将值w插入到二叉查找树中,这里的&表示引用。
{
if(root==NULL)
{
root=new node; //root=new node();
root->data=w;
root->left=root->right=NULL;
return;
}
if(w < root->data) insert(root->left,w);
else insert(root->right,w);
}
当然,此题引入了一个与二叉平衡树的类似概念-二叉查找树的镜像。
应该容易知道,在相同次序访问这两棵树时,只是改变了访问左右子树的顺序而已。比如说,前序序列(NLR)访问一个二叉查找树的镜像,那么等效于以(NRL)的顺序访问该二叉查找树。
那么,根据题目的要求,判断输入序列是不是某棵二叉查找树的前序序列或者二叉查找树镜像的前序序列就很容易了。
//只输出键值,故只需初始化一个Vector<int>变长整形数组
void preorder(node *root) //前序遍历
{
if(root==NULL)
return;
ans.push_back(root->data);
preorder(root->left);
preorder(root->right);
}
void postorder(node *root) //后序遍历
{
if(root==NULL)
return;
postorder(root->left);
postorder(root->right);
ans.push_back(root->data);
}
参考代码:
#include<iostream>
#include<cstdio>
#include<vector>
using namespace std;
const int inf = 1e9;
const int N = 1005;
struct node
{
int data;
node *left,*right;
};
node *root = NULL;
int a[N], n;
vector<int> ans;
void insert(node* &root,int w) //建二叉查找树
{
if(root==NULL)
{
root=new node; //root=new node();
root->data=w;
root->left=root->right=NULL;
return;
}
if(w < root->data) insert(root->left,w);
else insert(root->right,w);
}
void preorder(node *root) //前序遍历
{
if(root==NULL)
return;
ans.push_back(root->data);
preorder(root->left);
preorder(root->right);
}
void postorder(node *root) //后序遍历
{
if(root==NULL)
return;
postorder(root->left);
postorder(root->right);
ans.push_back(root->data);
}
void postorder2(node *root)
{
if(root==NULL)
return;
postorder2(root->right);
postorder2(root->left);
ans.push_back(root->data);
}
void preorder2(node *root)
{
if(root==NULL)
return;
ans.push_back(root->data);
preorder2(root->right);
preorder2(root->left);
}
bool check()
{
for(int i=0;i<ans.size();i++)
if(ans[i] != a[i]) return false;
return true;
}
void print()
{
for(int i=0;i<ans.size();i++)
{
if(i) cout<<" ";
cout<<ans[i];
}
}
int main()
{
cin>>n;
for(int i=0;i<n;i++)
cin>>a[i],insert(root,a[i]);
preorder(root);
if(check())
{
ans.clear();
printf("YES\n");
postorder(root);
print();
return 0;
}
ans.clear();
preorder2(root);
if(check())
{
ans.clear();
printf("YES\n");
postorder2(root);
print();
return 0;
}
cout<<"NO"<<endl;
return 0;
}