PAT-A1043 Is It a Binary Search Tree (25 分)不建树的DFS市面最简洁之一

mac2026-02-01  2

A1043题目地址:https://pintia.cn/problem-sets/994805342720868352/problems/994805440976633856

是之前好几道不建树版DFS求后序序列(A1086、A1138)的变型。

参考了liuchuo的代码:https://www.liuchuo.net/archives/2153

其中的这段代码还能更加精简:

getpost(0, n - 1); if(post.size() != n) { isMirror = true; post.clear(); getpost(0, n - 1); }

其实用下面这一行就能判断题目给的是BST还是BST的mirror了:

if(pre[0]>pre[1])dfs(0,n-1,true);//是BST else dfs(0,n-1,false);//是BST-mirror

整道题的AC代码如下: 

#include<iostream> #include<vector> using namespace std; int n,pre[1005]; vector<int> post; void dfs(int s,int e,bool flag){ int i=s+1,j=e; if(flag){ while(i<=e && pre[s]>pre[i])i++; while(j>s && pre[s]<=pre[j])j--; }else{ while(i<=e && pre[s]<=pre[i])i++; while(j>s && pre[s]>pre[j])j--; } if(i-j!=1)return; dfs(s+1,j,flag); dfs(i,e,flag); post.push_back(pre[s]); } int main(){ cin>>n; for(int i=0;i<n;i++)cin>>pre[i]; if(pre[0]>pre[1])dfs(0,n-1,true); else dfs(0,n-1,false); if(post.size()!=n)cout<<"NO"<<endl; else{ cout<<"YES"<<endl<<post[0]; for(int i=1;i<n;i++)cout<<" "<<post[i]; } return 0; }

 

最新回复(0)