PTA团体程序设计天梯赛-练习集 L3-010 是否完全二叉搜索树
原题链接
思路:
1.如果当前节点储存在
a
[
i
]
a[i]
a[i],那么他的左儿子储存在
a
[
i
∗
2
]
a[i*2]
a[i∗2],右儿子储存在
a
[
i
∗
2
+
1
]
a[i*2+1]
a[i∗2+1]。
反推过去也是成立的,当前节点存储在
a
[
i
]
a[i]
a[i],他的父亲节点为
a
[
i
/
2
]
a[i/2]
a[i/2]。
2.如何判断是完全二叉树:
顺序存储里,如果该树和满二叉树的节点编号一一对应,该树就是完全二叉树。
3.对于插入:递归插入,如果当前节点没有值,则插入到当前节点;如果待插入值大于当前节点的值,则递归到左子树,否则,递归到右子树。
4.对于输出:如果该二叉树不是完全二叉树的话,给出的n个节点不会填满前n个节点,所以在输出的时候要判断一下,如果值为
−
1
-1
−1的话直接跳过该节点,直到输出n个节点跳出循环。顺序存储的层次遍历就是按照下标大小的顺序。
n
n
n只有
20
20
20,完全不用担心内存超限什么的
代码:
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll,ll>PLL;
typedef pair<int,int>PII;
#define I_int ll
#define modl 19260817*19890604-19491001
inline ll read()
{
ll x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
#define read read()
char F[200];
inline void out(I_int x) {
if (x == 0) return (void) (putchar('0'));
I_int tmp = x > 0 ? x : -x;
if (x < 0) putchar('-');
int cnt = 0;
while (tmp > 0) {
F[cnt++] = tmp % 10 + '0';
tmp /= 10;
}
while (cnt > 0) putchar(F[--cnt]);
//cout<<" ";
}
const int maxn=1e5+100;
int a[maxn];
void dfs(int u,int x){
if(a[u]==-1){
a[u]=x;return ;
}
if(x>a[u]) dfs(u*2,x);
else dfs(u*2+1,x);
}
int b[maxn];
int main(){
memset(a,-1,sizeof a);
int n=read;
for(int i=1;i<=n;i++){
int x=read;
dfs(1,x);
}
int cnt=1,idx=1;
while(1){
if(a[idx]!=-1){
cout<<a[idx];
b[idx]=idx;
cnt++;
if(cnt==n+1) puts("");
else cout<<" ";
}
idx++;
if(cnt==n+1) break;
}
for(int i=1;i<=n;i++)
if(i!=b[i]){
puts("NO");
return 0;
}
puts("YES");
return 0;
}