天梯赛 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;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值