题意:给你一个二叉树的前序遍历序列,让你判断是不是二叉搜索树(左子树小于根,右子树大于根)或其镜像。如果是,输出后序遍历序列。
开始走了弯路,一直在思考如何建树,结果试了很多种方法都不对,总是在一个新节点该放在哪个节点下出问题。
其实这道题可以和给出前序中序写后序这种题类比,完全可以不建树。根据前序遍历的性质每次可以把根节点分离出来;根据二叉搜索树的性质每次可以把左右子树分开;这样在判断的时候只判断左边的是不是都小于根,右边的是不是都大于根就行了。至于后续序列,在递归左右之后,用b数组保存。
#include <stdio.h>
#include <algorithm>
using namespace std;
int a[1003], n, k = 0, b[1003];
int check1(int l, int r)
{
if(l > r)
return 1;
int root = a[l];//把根节点拿出来
int i, j;
for(i = l + 1 ; i <= r && a[i] < root; i++);//根据二叉搜索树的性质,把左右子树分开
for(j = i ; j <= r ; j++)//检查右子树是不是都大于根
{
if(a[j] < root)
return 0;
}
if(check1(l + 1, i - 1) == 0)//递归地检查左右子树
return 0;
if(check1(i, r) == 0)
return 0;
b[k++] = a[l];//b保存后序遍历序列
return 1;
}
int check2(int l, int r)
{
if(l > r)
return 1;
int root = a[l];
int i, j;
for(i = l + 1 ; i <= r && a[i] >= root; i++);
for(j = i ; j <= r ; j++)
{
if(a[j] >= root)
return 0;
}
if(check2(l + 1, i - 1) == 0)
return 0;
if(check2(i, r) == 0)
return 0;
b[k++] = a[l];
return 1;
}
int main()
{
int i;
scanf("%d", &n);
for(i = 1 ; i <= n ; i++)
scanf("%d", &a[i]);
if(n == 1)
{
printf("YES\n%d\n", a[1]);
return 0;
}
if(a[1] > a[2])
{
if(check1(1, n))
{
printf("YES\n");
for(i = 0 ; i < n - 1 ; i++)
printf("%d ", b[i]);
printf("%d\n", b[n - 1]);
}
else
printf("NO\n");
}
else
{
if(check2(1, n))
{
printf("YES\n");
for(i = 0 ; i < n - 1 ; i++)
printf("%d ", b[i]);
printf("%d\n", b[n - 1]);
}
else
printf("NO\n");
}
return 0;
}