PAT A1099 Build A Binary Search Tree (30 分) 二叉搜索树

最新推荐文章于 2022-01-25 18:17:54 发布

HackingGate

最新推荐文章于 2022-01-25 18:17:54 发布

阅读量101

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT

本文链接：https://blog.youkuaiyun.com/chch1996/article/details/99775409

PAT 专栏收录该内容

165 篇文章

订阅专栏

本文介绍了一种基于给定BST结构和元素序列构建平衡搜索树的方法。利用BST中序序列有序特性，通过修改中序遍历函数，将序列值赋给树节点。最终，通过层次遍历展示构建的BST。

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

题目大意：给定BST的结构，并给出其中元素的序列，构造出吻合的BST。

和A1064非常相似，还是利用BST中序序列有序的性质。由于题目给结构是类似于数组的格式，以-1代表NULL， 0代表根节点。所以也用数组来存储这棵树的节点，不过因为不是完全二叉树，只能通过在结构体中增加左右孩子域来记录左右孩子的信息。然后就是和A1064一样，改写中序遍历函数，在中序遍历到某一节点时，按顺序将中序序列的值赋给这个节点的data域。中序遍历的条件是节点不为空，这里就要分别判断下左右孩子的下标是否为-1再决定是否可以继续遍历。return的条件是中序序列所有的值都已经赋完了。

AC代码如下：

#include <iostream>
#include <vector>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <stack>
#include <queue>
#include <set>
#include <map>
#include <cstring>

using namespace std;

struct Node
{
    int data;
    int left;
    int right;
};

void inOrder(vector<int> &in, vector<Node> &tree, int parent)
{
    static int i = 0;
    if(i == in.size()) return;
    if(tree[parent].left > -1) inOrder(in, tree, tree[parent].left);
    tree[parent].data = in[i++];
    if(tree[parent].right > -1) inOrder(in, tree, tree[parent].right);
}

void levelOrder(vector<Node> &tree, int n)
{
    queue<int> q;
    q.push(0);
    int printCount = 0;
    while(!q.empty())
    {
        int front = q.front();
        q.pop();
        printf("%d", tree[front].data);
        printCount++;
        if(printCount < n) printf(" ");
        if(tree[front].left > -1) q.push(tree[front].left);
        if(tree[front].right > -1) q.push(tree[front].right);
    }
}

int main()
{
    int N;
    cin >> N;
    vector<Node> tree(N);
    vector<int> v(N);
    for (int i = 0; i < N; ++i)
    {
        scanf("%d%d", &(tree[i].left), &(tree[i].right));
    }
    for (int i = 0; i < N; ++i)
    {
        scanf("%d", &v[i]);
    }
    sort(v.begin(), v.end());
    inOrder(v, tree, 0);
    levelOrder(tree, N);
    return 0;
}