L2-004 这是二叉搜索树吗? (25 分)
一棵二叉搜索树可被递归地定义为具有下列性质的二叉树:对于任一结点,
- 其左子树中所有结点的键值小于该结点的键值;
- 其右子树中所有结点的键值大于等于该结点的键值;
- 其左右子树都是二叉搜索树。
所谓二叉搜索树的“镜像”,即将所有结点的左右子树对换位置后所得到的树。
给定一个整数键值序列,现请你编写程序,判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。
输入格式:
输入的第一行给出正整数 N(≤)。随后一行给出 N 个整数键值,其间以空格分隔。
输出格式:
如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果,则首先在一行中输出 YES
,然后在下一行输出该树后序遍历的结果。数字间有 1 个空格,一行的首尾不得有多余空格。若答案是否,则输出 NO
。
输入样例 1:
7
8 6 5 7 10 8 11
输出样例 1:
YES
5 7 6 8 11 10 8
输入样例 2:
7
8 10 11 8 6 7 5
输出样例 2:
YES
11 8 10 7 5 6 8
输入样例 3:
7
8 6 8 5 10 9 11
输出样例 3:
NO
因为前面写了二叉树的,所以这道题打算也用那种思路,写是写出来了,但是感觉怪怪的。
我建树是保存的下标编号,因为数可能是一样的,但是下标编号是唯一的。
建完树之后,写了个bfs搜了一下左右节点,发现好像也没毛病,但是自己想的一个样例过不了,肯定是错的代码,应该是后台样例比较水,所以满分水过了。。。
代码写的奇丑无比,但是还是希望有好心人能帮我想想我的代码哪里要改才能改对,过不了的样例:
2
2 1
输出应该为
YES
1 2
我的代码跑不出来。
后面看了一下别人的题解,发现还是别人的思路更简洁好实现一些,学到了,我这个弟弟。
先贴一下我的代码,不要学我的代码,有bug的,找不出来什么问题。。。
代码:
1 //L2-004 这是二叉搜索树吗? 2 #include<bits/stdc++.h> 3 using namespace std; 4 typedef long long ll; 5 const int maxn=1e5+10; 6 const int inf=0x3f3f3f3f; 7 8 int pre[maxn]; 9 int n,flag=0; 10 11 struct node{ 12 int l,r; 13 }tree[maxn]; 14 15 int buildmin(int l,int r) 16 { 17 if(l>r) return 0; 18 if(r==-1) return 0; 19 int rt=pre[l]; 20 int p1=r; 21 while(pre[p1]>=rt){ 22 if(p1==l) break; 23 p1--; 24 } 25 int minn=inf,pos; 26 for(int i=p1;i>l;i--){ 27 if(minn>pre[i]){ 28 minn=pre[i]; 29 pos=i; 30 } 31 } 32 for(int i=pos;i>l;i--){ 33 if(pre[i]>=rt){ 34 flag=1;break; 35 } 36 } 37 if(flag==1){ 38 buildmin(0,-1); 39 } 40 // cout<<pre[p1]<<endl; 41 else{ 42 if(p1!=l){ 43 tree[l].l=buildmin(l+1,p1); 44 tree[l].r=buildmin(p1+1,r); 45 } 46 else{ 47 tree[l].r=buildmin(p1+1,r); 48 } 49 return l; 50 } 51 } 52 53 int buildmax(int l,int r) 54 { 55 if(l>r) return 0; 56 if(r==-1) return 0; 57 int rt=pre[l]; 58 int p1=r; 59 while(pre[p1]<rt){ 60 p1--; 61 } 62 for(int i=p1;i>l;i--){ 63 if(pre[i]<rt){ 64 flag=1;break; 65 } 66 } 67 if(flag==1){ 68 buildmax(0,-1); 69 } 70 // cout<<pre[p1]<<endl; 71 else{ 72 if(p1!=r){ 73 tree[l].l=buildmax(l+1,p1); 74 tree[l].r=buildmax(p1+1,r); 75 } 76 else{ 77 tree[l].l=buildmax(l+1,p1); 78 } 79 return l; 80 } 81 } 82 83 void bfs(int x) 84 { 85 vector<int> vec; 86 queue<int> que; 87 que.push(x); 88 while(!que.empty()){ 89 int ret=que.front(); 90 que.pop(); 91 if(ret==0) break; 92 vec.push_back(ret); 93 if(tree[ret].l!=0){ 94 que.push(tree[ret].l); 95 } 96 if(tree[ret].r!=0){ 97 que.push(tree[ret].r); 98 } 99 } 100 int l=vec.size(); 101 for(int i=0;i<l;i++) 102 cout<<pre[vec[i]]<<" "; 103 cout<<endl; 104 } 105 106 vector<int> vec; 107 108 int behorder(int x) 109 { 110 if(tree[x].l==0&&tree[x].r==0){ 111 if(pre[x]!=0) 112 vec.push_back(pre[x]); 113 return 0; 114 } 115 behorder(tree[x].l); 116 behorder(tree[x].r); 117 vec.push_back(pre[x]); 118 } 119 120 void print() 121 { 122 int l=vec.size(); 123 for(int i=0;i<l-1;i++) 124 cout<<vec[i]<<" "; 125 cout<<vec[l-1]<<endl; 126 } 127 128 int main() 129 { 130 scanf("%d",&n); 131 for(int i=1;i<=n;i++) 132 scanf("%d",&pre[i]); 133 // cout<<"23333333333"<<endl; 134 if(pre[2]<pre[1]){ 135 buildmin(1,n); 136 if(flag==1) cout<<"NO"<<endl; 137 else{ 138 // bfs(1); 139 cout<<"YES"<<endl; 140 behorder(1); 141 print(); 142 } 143 } 144 else{ 145 buildmax(1,n); 146 if(flag==1) cout<<"NO"<<endl; 147 else{ 148 // bfs(1); 149 cout<<"YES"<<endl; 150 behorder(1); 151 print(); 152 } 153 } 154 // for(int i=1;i<=n;i++) 155 // cout<<tree[i].l<<" "<<tree[i].r<<endl; 156 // 157 //// else buildmax(1,n); 158 // behorder(); 159 160 }
学的别人的思路:
代码:
1 //L2-004 这是二叉搜索树吗? 2 #include<bits/stdc++.h> 3 using namespace std; 4 typedef long long ll; 5 const int maxn=1e5+10; 6 const int inf=0x3f3f3f3f; 7 8 int n; 9 int isMirror=0; 10 int pre[maxn]; 11 vector<int> vec; 12 13 int build(int l,int r) 14 { 15 if(l>r) return 0; 16 int rt=pre[l]; 17 int p1=l+1,p2=r; 18 if(!isMirror){ 19 while(p1<=r&&rt>pre[p1]) p1++; 20 while(p2>l&&rt<=pre[p2]) p2--; 21 } 22 else{ 23 while(p1<=r&&rt<=pre[p1]) p1++; 24 while(p2>l&&rt>pre[p2]) p2--; 25 } 26 if(p1-p2!=1) return 0; 27 build(l+1,p2);//按照后序遍历建树 28 build(p1,r); 29 vec.push_back(rt); 30 } 31 32 int main() 33 { 34 scanf("%d",&n); 35 for(int i=1;i<=n;i++) 36 scanf("%d",&pre[i]); 37 build(1,n); 38 if(vec.size()!=n){ 39 isMirror=1; 40 vec.clear(); 41 build(1,n); 42 } 43 if(vec.size()==n){ 44 printf("YES\n"); 45 int l=vec.size(); 46 for(int i=0;i<l-1;i++) 47 cout<<vec[i]<<" "; 48 cout<<vec[l-1]<<endl; 49 } 50 else{ 51 cout<<"NO"<<endl; 52 } 53 }
先这样。