天梯赛 L2-004 这是二叉搜索树吗?
正常来说需要中序再加上前序或后序才能建树,但这是一个搜索树,我们可以发现每一个节点的左子树都小于它,右子树都大于它,我们先不考虑镜像,镜像就是条件变一下,所以我们可以对先序遍历递归判断,对于每一个点,找到第一个大于它的那个点l,再判断从这个l+1到r是否都大于它,满足就是一个二叉搜索树,后序遍历也是递归的同时求得,后续还会加强对树的前序、中序、后序遍历的学习。
#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
typedef long long ll;
const int N = 1010;
const int mod = 998244353;
int n;
int tree[N];
int ans1[N],ans2[N],cnt1=0,cnt2=0;//存储后序结果
bool check1(int l,int r){ //判断是不是二叉搜索树
//递归的判断l,r
int x = tree[l];
if(l > r) return 1;
if(l == r){
ans1[cnt1++] = x;
return 1;
}
int i;
//找到当前子树左端点
for(i = l+1; i <= r && tree[i] < x ; i++) ;
i--;//左子树的最后一个
//找到当前子树右端点
int j;
for(j = i+1;j <= r && tree[j] >= x ; j++) ;
j--;
if(j != r) return 0;
bool flag = check1(l+1,i) & check1(i+1,r); //先访问左后右
ans1[cnt1++] = x;
return flag;
}
bool check2(int l,int r){ //判断是不是镜像二叉搜索树
//递归的判断l,r
int x = tree[l];
if(l > r) return 1;
if(l == r){
ans2[cnt2++] = x;
return 1;
}
int i;
//找到当前子树左端点
for(i = l+1; i <= r && tree[i] >= x ; i++) ;
i--;//左子树的最后一个
//找到当前子树右端点
int j;
for(j = i+1;j <= r && tree[j] < x ; j++) ;
j--;
if(j != r) return 0;
bool flag = check2(l+1,i) & check2(i+1,r);
ans2[cnt2++] = x;
return flag;
}
void solve() {
cin>>n;
for(int i = 0; i<n;i++){
cin>>tree[i];
}
if(check1(0,n-1)){
cout<<"YES"<<endl;
for(int i = 0 ; i < cnt1 ; i++){
if(i) cout<<" ";
cout<<ans1[i];
}
}else if(check2(0,n-1)){
cout<<"YES"<<endl;
for(int i = 0 ; i < cnt2 ; i++){
if(i) cout<<" ";
cout<<ans2[i];
}
}else{
cout<<"NO"<<endl;
}
}
signed main() {
ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
int tt = 1;
// cin >> tt;
while (tt--) {
solve();
}
return 0;
}