题目链接:点击打开链接
思路:按照BST的前序遍历和后序遍历建树,在建树的过程中判断出该BST是否唯一。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
using namespace std;
int pre[50],post[50];
int flag;
int root,le[50],rig[50];
vector<int> v;
void buildTree(int &ro,int l1,int r1,int l2,int r2){//pre post
if(l1 > r1 || l2 > r2) return;
ro = l1;
if(l1 == r1) return;
int te = pre[l1 + 1],loc;
for(loc = l2;post[loc] != te;loc++);
int cnt = (r2 - l2) - (loc - l2 + 1);
if(!cnt){//当前节点只有一个子树时,无法确定该子树为左子树还是右子树,从而判断出该BST不唯一
flag = 0;
buildTree(le[ro],l1 + 1,r1,l2,r2 - 1);//方便起见,不确定的子树建为左子树
}
else{
buildTree(le[ro],l1 + 1,l1 + 1 + loc - l2,l2,loc);
buildTree(rig[ro],l1 + 2 + loc - l2,r1,loc + 1,r2 - 1);
}
}
void inTravel(int r){
if(!r) return;
inTravel(le[r]);
v.push_back(r);
inTravel(rig[r]);
}
int main(){
int n;
flag = 1;
memset(le,0,sizeof(le));
memset(rig,0,sizeof(rig));
scanf("%d",&n);
for(int i = 1;i <= n;i++){
scanf("%d",&pre[i]);
}
for(int i = 1;i <= n;i++){
scanf("%d",&post[i]);
}
buildTree(root,1,n,1,n);
inTravel(root);
puts(flag?"Yes":"No");
for(int i = 0;i < v.size();i++){
printf("%d%c",pre[v[i]],i == v.size() - 1?'\n':' ');
}
return 0;
}