题目链接:https://pintia.cn/problem-sets/994805046380707840/problems/994805070971912192
L2-004 这是二叉搜索树吗? (25 分)
一棵二叉搜索树可被递归地定义为具有下列性质的二叉树:对于任一结点,
- 其左子树中所有结点的键值小于该结点的键值;
- 其右子树中所有结点的键值大于等于该结点的键值;
- 其左右子树都是二叉搜索树。
所谓二叉搜索树的“镜像”,即将所有结点的左右子树对换位置后所得到的树。
给定一个整数键值序列,现请你编写程序,判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。
输入格式:
输入的第一行给出正整数 N(≤1000)。随后一行给出 N 个整数键值,其间以空格分隔。
输出格式:
如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果,则首先在一行中输出
YES
,然后在下一行输出该树后序遍历的结果。数字间有 1 个空格,一行的首尾不得有多余空格。若答案是否,则输出NO
。输入样例 1:
7 8 6 5 7 10 8 11
输出样例 1:
YES 5 7 6 8 11 10 8
输入样例 2:
7 8 10 11 8 6 7 5
输出样例 2:
YES 11 8 10 7 5 6 8
输入样例 3:
7 8 6 8 5 10 9 11
输出样例 3:
NO
解题思路:考查二叉搜索树的知识。。。
递归模拟先序遍历的过程,关键是判断节点的左右子树的分界点,
因为二叉搜索树中,对于任一节点,其左子树中所有结点的键值小于该结点的键值,
其右子树中所有结点的键值大于等于该结点的键值,其左右子树都是二叉搜索树,
所以通过大小关系可以找到分界点,同时判断是否满足上述条件。 具体实现见代码(有详细注释)。
/*
考查二叉搜索树的知识。。。
递归模拟先序遍历的过程,关键是判断节点的左右子树的分界点,
因为二叉搜索树中,对于任一节点,其左子树中所有结点的键值小于该结点的键值,
其右子树中所有结点的键值大于等于该结点的键值,其左右子树都是二叉搜索树,
所以通过大小关系可以找到分界点,同时判断是否满足上述条件。
*/
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <set>
#include <stack>
#include <queue>
#include <vector>
#include <string>
#include <utility>
#define cla(a, sum) memset(a, sum, sizeof(a))
#define rap(i, m, n) for(int i=m; i<=n; i++)
#define rep(i, m, n) for(int i=m; i>=n; i--)
#define bug printf("???\n")
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> P;
typedef pair<ll, ll> Pl;
const int Inf = 0x3f3f3f3f;
const double eps = 1e-8;
const int mod=998244353;
const int maxn = 1e3+5;
const int N=1e5;
template <typename T> void read(T &x){
x = 0; int f = 1; char ch = getchar();
while (!isdigit(ch)) {if (ch == '-') f = -1; ch = getchar();}
while (isdigit(ch)) {x = x * 10 + ch - '0'; ch = getchar();}
x *= f;
}
int n,a[maxn];
struct node{
int val;
node *l,*r;
};
int flag=0;//标记,判断是否满足二叉搜索树或其镜像
node *buildmin(int *p,int q){//p为键值数组,q为数组大小
if(q<=0)return NULL;
node *u=new node();//建立新节点
int i;
u->val =p[1];
for(i=2;i<=q;i++){//找到第一个大于或等于的键值,即分界点
if(p[i]>=p[1])break;
}
for(int j=i;j<=q;j++){//后面的键值均大于或等于才符合,反之不符合
if(p[j]<p[1]){
flag=1;
return NULL;
}
}
//分别建立左右子树 ,注意参数的更新
u->l =buildmin(p+1,i-2);
u->r =buildmin(p+i-1,q-i+1);
return u;
}
node * buildmax(int *p,int q){//同理buildmin,判断是否符合镜像
if(q<=0)return NULL;
node *u=new node();
int i;
u->val =p[1];
for(i=2;i<=q;i++){
if(p[i]<p[1])break;
}
for(int j=i;j<=q;j++){
if(p[j]>=p[1]){
flag=1;
return NULL;
}
}
u->l =buildmax(p+1,i-2);
u->r =buildmax(p+i-1,q-i+1);
return u;
}
void print(node *u,int tm){//后序遍历
if(u->l )print(u->l ,tm+1);
if(u->r )print(u->r ,tm+1);
//注意输出格式
if(tm!=1)
printf("%d ",u->val );
else printf("%d",u->val );
}
int main()
{
//ios::sync_with_stdio(false),cin.tie(0);
cin>>n;
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
}
//先判断是否满足二叉搜索树
node *temp;
temp=buildmin(a,n);
if(flag==0){
printf("YES\n");
print(temp,1);
return 0;
}
//再判断是否满足镜像
node *tmp;flag=0;
tmp=buildmax(a,n);
if(flag==0){
printf("YES\n");
print(tmp,1);
}
else printf("NO");
return 0;
}