L2-004 CCCC练习

链接：https://www.patest.cn/contests/gplt/L2-004

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

直接模拟即可建树
参考博客：https://blog.csdn.net/weixin_37517391/article/details/64592056

#include <bits/stdc++.h>
using namespace std;
#define me(a,b) memset(a,b,sizeof(a));
bool flag;
int tre[1100];
int lef[1100];
int rig[1100];
int sum;
int  dfs_1(int l,int r,int *tree)
{
    if(l>r)  return 0;
    int p=l+1;
    while(p<=r&&tree[p]<=tree[l]) p++;
    for(int i=p;i<=r;i++)
    {
        if(tree[l]>tree[i])
        {   

            flag=true; 
            return l;
        }

    }
    lef[l]=dfs_1(l+1,p-1,tree);
    rig[l]=dfs_1(p,r,tree);
    return l;




}
int  dfs_2(int l,int r,int *tree)
{
    if(l>r)  return 0;
    int p=l+1;
    while(p<=r&&tree[p]>=tree[l]) p++;
    for(int i=p;i<=r;i++)
    {
        if(tree[l]<tree[i])
        {   

            flag=true; 
            return l;
        }

    }
    lef[l]=dfs_2(l+1,p-1,tree);
    rig[l]=dfs_2(p,r,tree);
    return l;




}
int ff;
void print(int x)
{
    if(lef[x]) print(lef[x]);
    if(rig[x]) print(rig[x]);
    if(!ff) cout<<" ";
    else ff=0;
    cout<<tre[x];
}
int main()
{

    int n;
    cin>>n;
    for(int i=0;i<n;i++)
    cin>>tre[i];
    flag=false; 
    dfs_1(0,n-1,tre);
        ff=1;
    if(!flag)
    {
        cout<<"YES"<<endl;

        print(0);
    }
    else
    {
        flag=false;
        dfs_2(0,n-1,tre);
        if(!flag)
        {
            cout<<"YES"<<endl;
            print(0);
        }
        else
        {
            cout<<"NO"<<endl;
        }

    }
    return 0;
}

猜你喜欢