PTA-L2-004 这是二叉搜索树吗? (25分)
这道题树的问题。
给出了前序遍历,问我们通过这个判断这个树是不是二叉搜索树的前序遍历或者这棵树镜像的前序遍历。是的话输出其后序遍历,不是的话输出“NO"
根据二叉搜索树及前序遍历的特点,我们可以得知一个前序遍历中,第一个肯定是根节点,那么我们就要找出左子树和右子树的分界位置,找的方法就是根据二叉搜索树的特点,左子树都小于根节点,右子树都大于等于根节点,用指针过来索引。
并且这两个指针的差满足等于1.
因为是分界线的位置,不可能这两个指针的中间还有数字把。
如果不等于1,说明不是二叉搜索树的前序遍历。
然后继续递归找下去。
我们在递归的时候同时把后续遍历存起来。
后续遍历是左右根,所以在遍历完左右子树的时候,才进行存储。
我们在找的时候特别注意边界问题:
从根节点的下一个结点区分左右子树,递归的时候分别递归左子树和右子树。
其中用指针找分界点的位置的时候:我们还要保证这两个指针不能越界,因为在后面的递归中要从根节点的下一个结点进行操作,所以从右往左的指针的边界条件是大于根节点的位置。
如果写大于等于的话,那么从右往左的指针越界了,使得当只有一个结点的树时在两个指针的差满足等于1的时候判断错误。
简单来说,我们只能有一边的指针可以判断等于边界情况。
上述情况是左边的指针可以等于边界情况,右边的指针不行。
(主要判断的是只有一个结点的树的情况)
这样就可以跟两个指针的差满足等于1对应起来了,否则会判断错误。
同理,判断前序遍历不是二叉搜索树的前序遍历,需要进一步判断这个前序遍历是否是镜像二叉搜索树的前序遍历。
直接在递归里面加一个条件即可。
镜像的二叉搜索树的左子树大于等于根节点,
右子树小于根节点
如果都不是,那么就输出NO了。
如果是那就输出二叉搜索树/镜像二叉搜索树的后序遍历了。
代码部分:
#include <bits/stdc++.h>
#define mst(a, n) memset(a, n, sizeof(a))
using namespace std;
const int N = 1e3 + 10;
const int INF = 1e6 + 10;
typedef long long ll;
int n;
int a[N];
int flag;
int ans[N];
int cnt;
int f;
void solve(int l, int r)
{
if (l > r)
{
return ;
}
int tl = l + 1;
int tr = r;
if (!flag)
{
while (tl <= r && a[tl] < a[l])
{
tl++;
}
while (tr > l && a[tr] >= a[l])
{
tr--;
}
}
else
{
while (tl <= r && a[tl] >= a[l])
{
tl++;
}
while (tr > l && a[tr] < a[l])
{
tr--;
}
}
if (tl - tr != 1)
{
return ;
}
solve(l + 1, tr);
solve(tl, r);
ans[cnt++] = a[l];
}
int main()
{
cin >> n;
for (int i = 0; i < n; i++)
{
cin >> a[i];
}
solve(0, n - 1);
if (cnt == n)
{
f = 1;
}
else
{
cnt = 0;
flag = 1;
solve(0, n - 1);
if (cnt == n)
{
f = 1;
}
}
if (!f)
{
cout << "NO\n";
}
else
{
cout << "YES\n";
for (int i = 0; i < cnt; i++)
{
if (i)
{
cout << " ";
}
cout << ans[i];
}
cout << endl;
}
return 0;
}