PTA L2-004 这是二叉搜索树吗？

题目

一棵二叉搜索树可被递归地定义为具有下列性质的二叉树：对于任一结点，其左子树中所有结点的键值小于该结点的键值；其右子树中所有结点的键值大于等于该结点的键值；其左右子树都是二叉搜索树。所谓二叉搜索树的“镜像”，即将所有结点的左右子树对换位置后所得到的树。给定一个整数键值序列，现请你编写程序，判断这是否是对一棵二叉搜索树或其镜像进行前序遍历的结果。

输入格式：

输入的第一行给出正整数 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

解题思路

搜索二叉树的性质，从左至右第一个大于等于根节点的下标比从右至左第一个小于根节点的下标大1.
镜像相反…

AC代码（1）

#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> m,h;
bool v;
void dfs(int l,int r)
{
    if(l>r)
        return;
    int i=l+1,j=r;
    if(v==true)
    {
        while(m[i]<m[l]&&i<=r)++i;
        while(m[j]>=m[l]&&j>l)--j;
    }
    else
    {
        while(m[i]>=m[l]&&i<=r)++i;
        while(m[j]<m[l]&&j>l)--j;
    }
    if(i-j!=1)
        return;
    dfs(l+1,j);
    dfs(i,r);
    h.push_back(m[l]);
}
int main()
{
    int n;
    scanf("%d",&n);
        m.resize(n);
        for(int i=0; i<n; ++i)
            scanf("%d",&m[i]);
        v=true;
        dfs(0,n-1);
        if(h.size()!=n)
        {
            v=false;
            h.clear();
            dfs(0,n-1);
        }
        if(h.size()==n)
        {
            printf("YES\n");
            for(int i=0; i<n; ++i)
            {
                if(i)
                    putchar(' ');
                printf("%d",h[i]);
            }
            putchar('\n');
        }
        else
            printf("NO\n");
}

AC代码（2)

#include<stdio.h>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> m,h;
bool v;
void dfs(int l,int r)
{
    if(l>r)
        return;
    int i=l+1,j=r;
    if(v==true)
    {
        while(m[i]<m[l]&&i<=r)++i;
        while(m[j]>=m[l]&&j>l)--j;
    }
    else
    {
        while(m[i]>=m[l]&&i<=r)++i;
        while(m[j]<m[l]&&j>l)--j;
    }
    if(i-j!=1)
        return;
    dfs(l+1,j);
    dfs(i,r);
    h.push_back(m[l]);
}
int main()
{
    int n;
    scanf("%d",&n);
        m.resize(n);
        for(int i=0; i<n; ++i)
            scanf("%d",&m[i]);
        v=true;
        if(m[0]<m[1])
            v=false;
        dfs(0,n-1);
        if(h.size()==n)
        {
            printf("YES\n");
            for(int i=0; i<n; ++i)
            {
                if(i)
                    putchar(' ');
                printf("%d",h[i]);
            }
            putchar('\n');
        }
        else
            printf("NO\n");
}