PTA 7-28 搜索树判断 (25 point(s))

最新推荐文章于 2023-06-11 01:15:08 发布

moleet

最新推荐文章于 2023-06-11 01:15:08 发布

阅读量556

点赞数

分类专栏： # 树文章标签：二叉搜索树树结构 vector 递归

本文链接：https://blog.csdn.net/Atl212/article/details/119674055

版权

树专栏收录该内容

5 篇文章

订阅专栏

该代码实现了一个程序，用于判断输入的整数序列是否为二叉搜索树或镜像二叉搜索树的前序遍历，并在正确的情况下输出对应的后序遍历序列。程序通过递归地检查输入序列是否满足二叉搜索树或其镜像的性质，并在找到匹配情况时生成并输出后序遍历序列。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

对于二叉搜索树，我们规定任一结点的左子树仅包含严格小于该结点的键值，而其右子树包含大于或等于该结点的键值。如果我们交换每个节点的左子树和右子树，得到的树叫做镜像二叉搜索树。

现在我们给出一个整数键值序列，请编写程序判断该序列是否为某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列，如果是，则输出对应二叉树的后序遍历序列。

输入格式:

输入的第一行包含一个正整数N（≤1000），第二行包含N个整数，为给出的整数键值序列，数字间以空格分隔。

输出格式:

输出的第一行首先给出判断结果，如果输入的序列是某棵二叉搜索树或某镜像二叉搜索树的前序遍历序列，则输出YES，否侧输出NO。如果判断结果是YES，下一行输出对应二叉树的后序遍历序列。数字间以空格分隔，但行尾不能有多余的空格。

输入样例1:

7
8 6 5 7 10 8 11

结尾无空行

输出样例1:

YES
5 7 6 8 11 10 8

结尾无空行

输入样例2:

7
8 6 8 5 10 9 11

输出样例2:

NO

#include <bits/stdc++.h>
using namespace std;
int key[1000];
vector<int> tree[2];

bool judge(int root, int top, bool m){
    // 左右子集下标
    int left = root + 1, right = top;
    if(m == true){
        // 镜像 先序输出(根右左)
        while(left <= top && key[left] >= key[root]) left++;
        while(right > root && key[right] < key[root]) right--;
    }
    else{
        // 非镜像(根左右)
        while(left <= top && key[left] < key[root]) left++;
        while(right > root && key[right] >= key[root]) right--;
    }
    // 类似二分查找 左右边界最后交叉
    if(left - right != 1)
        return false;
    // 继续对左右子集判断 由于下标交叉 左子集上界为top=right 同理右子集根root=left
    judge(root + 1, right, m);
    judge(left, top, m);
    // 后序输出
    tree[m].push_back(key[root]);
    return true;
}

int main() {
    int n;
    cin >> n;
    for(int i = 0; i < n; i++)
        cin >> key[i];
    for(int mirror = 1; mirror >= 0; mirror--){
        // BST判断 && 输出元素与输入元素数量相等
        if(judge(0, n-1, mirror) == true && tree[mirror].size() == n){
            cout << "YES" << endl << tree[mirror][0];
            for(int i = 1; i < n; i++)
                cout << " " << tree[mirror][i];
            exit(0);
        }
    }
    cout << "NO";
}