PAT4-06. 搜索树判断

先序遍历都是(根,左子树,右树),后序遍历改成(左子树,右子树,根)即可,判断是否为线序遍历和先序改为后序遍历都是用分治法。样例case都没问题但最后一个case没过,真奇怪。

#include<algorithm>
#include<iostream>
#include<vector>
using namespace std;
typedef vector<int>::iterator itr;
bool isbst(itr beg,itr end){
  if(end-beg<=2)return true;
  itr mid=find_if(beg+1,end,[&](int x)->bool{return x>=*beg;});
  if(any_of(mid,end,[&](int x)->bool{return x<*beg;}))
    return false;
  return isbst(beg+1,mid)&&isbst(mid,end);
}
bool ismbst(itr beg,itr end){
  if(end-beg<=2)return true;
  itr mid=find_if(beg+1,end,[&](int x)->bool{return x<*beg;});
  if(any_of(mid,end,[&](int x)->bool{return x>=*beg;}))
    return false;
  return ismbst(beg+1,mid)&&ismbst(mid,end);
}
void pre2post_bst(itr beg,itr end){
  if(end-beg<=1)return;
  itr mid=find_if(beg+1,end,[&](int x)->bool{return x>=*beg;});
  pre2post_bst(beg+1,mid);
  pre2post_bst(mid,end);
  rotate(beg,beg+1,end);
}
void pre2post_mbst(itr beg,itr end){
  if(end-beg<=1)return;
  itr mid=find_if(beg+1,end,[&](int x)->bool{return x<*beg;});
  pre2post_mbst(beg+1,mid);
  pre2post_mbst(mid,end);
  rotate(beg,beg+1,end);
}

int main(){
  int n;cin>>n;
  vector<int>val(n),ret;
  for(auto &x:val)cin>>x;
  if(isbst(val.begin(),val.end())){
    pre2post_bst(val.begin(),val.end());
    ret=val;}
  if(ismbst(val.begin(),val.end())){
    pre2post_mbst(val.begin(),val.end());
    ret=val;
  }
  if(ret.size()){
    bool first=true;
    cout<<"YES"<<endl;
    for(auto x:ret){
      if(first)first=false;
      else cout<<' ';
      cout<<x;}
  }else cout<<"NO"<<endl;
  return 0;
}



  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值