PTA-L2-004 这是二叉搜索树吗? (25分)

19 篇文章 0 订阅

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;
} 
  • 3
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

娃娃酱斯密酱

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值