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

题目大意

给一个二叉树的前序遍历，让你判断是否是二叉搜索树（左子树全部节点小于根节点，右子树所有节点大于等于根节点)或者是二叉搜索树的镜像。如果是，输出后序遍历

思路

对整个树先判断是否符合条件，然后递归判断子树。默认为不镜像 如果不镜像无法满足题意，那就镜像一下再dfs，如果还不满足，那就是NO

代码

include<bits/stdc++.h>
using namespace std;
int pre[1005];
vector<int>post;
bool fz=false;
void dfs(int root, int r)
{
    if(root>r) return; //return的结果是下次递归无法进行，且无法输入到post里 导致post.size小于n
    int i = root+1, j=r;//i是左子树全部节点的下标，r是右子树全部节点的下标
    if(!fz)
    {
        while(pre[i]<pre[root]&&i<=r) i++;//i最多到r+1（全单调情况） 这样r+1-r=1 符合题意
        while(pre[j]>=pre[root]&&j>root) j--;//j最多到root-1 也是全单调情况 符合题意
    }
    else
    {
        while(pre[i]>=pre[root]&&i<=r) i++;//镜像 大小要求刚好相反
        while(pre[j]<pre[root]&&j>root) j--;
    }
    if(i-j!=1) return;
    dfs(root+1,j);
    dfs(i,r);
    post.push_back(pre[root]);
}
int main()
{
    int n;
    cin>>n;
    for(int i =0;i<n;i++)
        cin>>pre[i];
    dfs(0,n-1);
    if(post.size()!=n)
    {
        post.clear();
        fz=true;
        dfs(0,n-1);
    }
    if(post.size()==n)
    {
        printf("YES\n");
        int f=1;
        for(auto it:post)
        {
            if(f)
            {
                f=0;
                cout<<it;
            }
            else
            {
                cout<<" "<<it;
            }
        }
        puts("");
    }
    else
        printf("NO\n");
}