L2-004 这是二叉搜索树吗？ （25 分）(二叉树的遍历)

【题解】

思路：一棵合法的二叉搜索树的先序遍历序列满足，对于区间[l,r]，存在一个点i，a[l]<a[i]<=a[r]。

递归判断下是否是合法的二叉搜索树或者镜像(a[l]>=a[i]>a[r])即可。

【代码1】

#include <bits/stdc++.h>
using namespace std;
vector <int> post;
int pre[1005];
bool isMirror=false;
void getpost(int l,int r)
{
	if(l>r) return ;
	int i=l+1,j=r;
	if(!isMirror)
	{
		while(i<=r&&pre[l]>pre[i]) i++; //从当前的根节点向后遍历
		while(j>l&&pre[l]<=pre[j]) j--; //从当前的尾节点向前遍历
	}
	else
	{
		while(i<=r&&pre[l]<=pre[i]) i++;
		while(j>l&&pre[l]>pre[j]) j--;
	}
	if(i-j!=1) return;
	getpost(l+1,j); //左子树
	getpost(i,r); //右子树
	post.push_back(pre[l]); //存放根节点
}
int main()
{
	int n; scanf("%d",&n);
    for(int i=0;i<n;i++) scanf("%d",&pre[i]);
    post.clear();
    getpost(0,n-1);
    if(post.size()!=n)
    {
        isMirror=true;
        post.clear();
        getpost(0,n-1);
    }
    if(post.size()!=n) puts("NO");
    else
    {
        puts("YES");
        for(int i=0;i<n;i++)
            printf(i!=n-1?"%d ":"%d\n",post[i]);
    }
    return 0;
}

【代码2】

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1005;
int a[maxn];
int tree[maxn];
struct Node{
	int val;
	Node *lchild;
	Node *rchild;
};
Node *root = NULL;
Node* build1(Node *t,int key)
{
	if(t==NULL){ //根结点
		Node *p = new Node;
		p->val = key;
		p->lchild = NULL;
		p->rchild = NULL;
		return p;
	}
	if(key >= t->val)
		t->rchild = build1(t->rchild,key);
	else
		t->lchild = build1(t->lchild,key);
	return t;
}
Node* build2(Node *t,int key)
{
	if(t==NULL)
	{
		Node*p = new Node;
		p->val = key;
		p->lchild = NULL;
		p->rchild = NULL;
		return p;
	}
	if(key < t->val)
		t->rchild = build2(t->rchild,key);
	else
		t->lchild = build2(t->lchild,key);
	return t;
}
void print(Node *t)
{
	if(t==NULL) return;
	print(t->lchild);
	print(t->rchild);
	printf(t!=root?"%d ":"%d\n",t->val);
}
int check1(int l,int r) //二叉搜索树
{
	if(l>=r) return 1;
	int key=r;
	for(int i=l+1;i<=r;i++)
		if(a[l]<=a[i]){
			key=i;
            break;
		}
	for(int i=key+1;i<=r;i++)
		if(a[l]>a[i])
			return 0;
	return check1(l+1,key-1)&&check1(key,r); //检验左右子树
}
int check2(int l,int r) //镜像
{
	if(l>=r) return 1;
	int key=r;
	for(int i=l+1;i<=r;i++)
		if(a[l]>a[i]){
			key=i;
			break;
		}
	for(int i=key+1;i<=r;i++)
		if(a[l]<=a[i])
			return 0;
	return check2(l+1,key-1)&&check2(key,r);
}
int main()
{
    int n; scanf("%d",&n);
	for(int i=0;i<n;i++) scanf("%d",&a[i]);
	if(check1(0,n-1))
		for(int i=0;i<n;i++)
			root=build1(root,a[i]);
	else if(check2(0,n-1))
		for(int i=0;i<n;i++)
			root=build2(root,a[i]);
    else{
        printf("NO");
        return 0;
    }
	printf("YES");
	print(root);
    return 0;
}

【题目】

L2-004 这是二叉搜索树吗？ （25 分）

一棵二叉搜索树可被递归地定义为具有下列性质的二叉树：对于任一结点，

• 其左子树中所有结点的键值小于该结点的键值；
• 其右子树中所有结点的键值大于等于该结点的键值；
• 其左右子树都是二叉搜索树。

所谓二叉搜索树的“镜像”，即将所有结点的左右子树对换位置后所得到的树。

给定一个整数键值序列，现请你编写程序，判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。

输入格式：

输入的第一行给出正整数 N（≤1000）。随后一行给出 N 个整数键值，其间以空格分隔。

输出格式：

如果输入序列是对一棵二叉搜索树或其镜像进行前序遍历的结果，则首先在一行中输出 YES ，然后在下一行输出该树后序遍历的结果。数字间有 1 个空格，一行的首尾不得有多余空格。若答案是否，则输出 NO。

输入样例 1：

7
8 6 5 7 10 8 11

输出样例 1：

YES
5 7 6 8 11 10 8

输入样例 2：

7
8 10 11 8 6 7 5

输出样例 2：

YES
11 8 10 7 5 6 8

输入样例 3：

7
8 6 8 5 10 9 11

输出样例 3：

NO