可达鸭 验证二叉搜索树 题解。
题目描述:
给你一棵二叉树的信息,请你判断一下这棵树是不是一棵有效的二叉搜索树。
输入输出描述
输入描述:
第一行输入一个整数
n
n
n,表示接下来要输入
n
n
n个数据;
第二行输入
n
n
n个数据
a
i
a_i
ai,第i个数字表示第i个节点的数值为
a
i
a_i
ai。
如果输入的为0,表示这个位置没有节点。节点个数不超过10000个,数据之前用空格间隔。
输出描述:
如果是一棵有效的二叉搜索树,输出”true”;
如果不是一棵二叉搜索树,输出”false”
输入输出样例
输入样例:
7
5 1 4 0 0 3 6
输出样例:
false
解题思路:
解题思路分三步:存储,求中序遍历,判断并输出。
1.存储树:使用结构体,设变量data为数据,L为左儿子,r为右儿子。
如果节点a[i]有左儿子,将左儿子设为a[i乘2]。
否则,将左儿子设为-1。
如果节点a[i]有右儿子,将右儿子设为a[i乘2+1]。
否则,将右儿子设为-1。
for(int i=1;i<=n;i++)
{
cin>>a[i];
}
for(int i=1;i<=n;i++)
{
tree[i].data=a[i];
if(a[i*2]!=0)
{
tree[i].l=a[i*2];
}
else tree[i].l=-1;
if(a[i*2+1]!=0)
{
tree[i].r=a[i*2+1];
}
else tree[i].r=-1;
}
(根据二叉树的性质,左儿子编号为节点编号乘2,右儿子编号为节点编号乘2+1)
2.中序遍历:遍历tree数组,
若节点有左儿子,则递归遍历左子树,
将该节点加入zhong数组,
若节点有右儿子,则递归遍历右子树。
void get_zhong(int x)
{
if(tree[x].data==0) return;
if(tree[x].l!-1)
{
get_zhong(2*x);
}
zhong[idx++]=tree[x].data;
if(tree[x].r!=-1)
{
get_zhong(x*2+1);
}
}
3.判断并输出:若中序遍历结果非递增(不符合二叉搜索树特点),输出false。
否则,输出true。
for(int i=1;i<=idx-1;i++)
{
if(zhong[i]<zhong[i-1])
{
cout<<"false";
return 0;
}
}
cout<<"true";
AC代码
#include<bits/stdc++.h>
using namespace std;
const int N=20005;
int n;
int idx=0;
struct node{
int data;
int l,r;
}tree[N];//结构体存树
int a[N];//输入数组
int zhong[N];//中序遍历数组
void get_zhong(int x)//求中序遍历
{
if(tree[x].data==0) return;//如果没有这个节点,返回
if(tree[x].l!-1)//有左儿子
{
get_zhong(2*x);//递归搜索左子树
}
zhong[idx++]=tree[x].data;//在中序遍历中保存该节点
if(tree[x].r!=-1)//有右儿子
{
get_zhong(x*2+1);//递归搜索右子树
}
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
cin>>a[i];
}//输入
for(int i=1;i<=n;i++)
{
tree[i].data=a[i];//保存节点
if(a[i*2]!=0)//有左儿子
{
tree[i].l=a[i*2];//保存左儿子
}
else tree[i].l=-1;//没左儿子,将左儿子设为-1
if(a[i*2+1]!=0)//有右儿子
{
tree[i].r=a[i*2+1];//保存右儿子
}
else tree[i].r=-1;//没右儿子,将右儿子设为-1
}
get_zhong(1);//递归求中序遍历
for(int i=1;i<=idx-1;i++)
{
if(zhong[i]<zhong[i-1])//如果中序遍历非递增
{
cout<<"false";//输出false
return 0;
}
}
cout<<"true";//输出ture
return 0;
}