L2-004这是二叉搜索树吗

最新推荐文章于 2024-08-23 18:53:20 发布

转载最新推荐文章于 2024-08-23 18:53:20 发布 · 154 阅读

算法专栏收录该内容

112 篇文章

订阅专栏

本文介绍了一种算法，用于判断给定的前序遍历序列是否对应于一棵二叉搜索树或其镜像。通过两次尝试构建后序遍历序列来验证，确保了结果的准确性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

转自点击打开链接

分析：假设它是二叉搜索树，一开始isMirror为FALSE，根据二叉搜索树的性质将已知的前序转换为后序，转换过程中，如果发现最后输出的后序数组长度不为n，那就设isMirror为true，然后清空后序数组，重新再转换一次（根据镜面二叉搜索树的性质），如果依旧转换后数组大小不等于n，就输出no否则输出yes

#include<cstdio>
#include<vector>
using namespace std;
bool isMirror;
vector<int>pre;
vector<int>post;
void getpost(int root,int tail){
   if(root>tail) return;
   int i=root+1,j=tail;
   if(!isMirror){
    while(i<=tail&&pre[root]>pre[i]) i++;
    while(j>root&&pre[root]<=pre[j]) j--;
   }else {
      while(i<=tail&&pre[root]<=pre[i])  i++;
      while(j>root&&pre[root]>pre[j]) j--;
   }
   if(i-j!=1) return ;
   getpost(root+1,j);
   getpost(i,tail);
   post.push_back(pre[root]);
}
int main(){
   int n;
   scanf("%d",&n);
   pre.resize(n);
   for(int i=0;i<n;i++){
    scanf("%d",&pre[i]);
   }
   getpost(0,n-1);
   if(post.size()!=n){
      isMirror=true;
      post.clear();
      getpost(0,n-1);
   }
   if(post.size()==n){
    printf("YES\n%d",post[0]);
    for(int i=1;i<n;i++)
        printf(" %d",post[i]);
    }else {
      printf("NO");
    }
    return 0;
}