【nlogn 构造二叉搜索树】CodeForces - 675D Tree Construction

最新推荐文章于 2024-08-14 21:49:21 发布

笑对这个世界的志贵

最新推荐文章于 2024-08-14 21:49:21 发布

阅读量334

点赞数

分类专栏：比赛卡住题数据结构二叉树排序平衡树

比赛卡住题同时被 2 个专栏收录

92 篇文章

订阅专栏

数据结构二叉树排序平衡树

9 篇文章

订阅专栏

本文介绍了一种通过使用C++实现的高效算法来构造二叉搜索树并查找指定节点的父节点的方法。通过利用STL中的set容器和lower_bound函数，该算法能够快速定位到目标节点的父节点。详细步骤包括初始化深度数组、插入节点、更新父节点和深度，最终输出所有节点的父节点。

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

Step1 Problem:

给你 n 个数，按照输入顺序构造二叉搜索树，然后输出每个数的父亲。

Step2 Ideas:

二叉搜索树：将要插入的数 num, 树上比 num 大的数 Max, 比 num 小的树 Min. 那么 Max 或者 Min 一定是 num 的父亲。
所以我们只用记录每个点的深度，然后求出 Max 和 Min 判断谁比较深，就可以了。
set 得用 std::set::lower_bound
std::lower_bound 对于 set 而言非常的慢

Step3 Code:

#include<bits/stdc++.h>
using namespace std;
const int N = 112345;
struct node
{
    int x, id;
    bool operator < (const node &b) const {
        return x < b.x;
    }
};
int fq[N], dep[N];
set<node> q;
set<node>::iterator lt, rt;
int main()
{
    int n, num;
    scanf("%d", &n);
    dep[0] = 0;
    for(int i = 0; i < n; i++)
    {
        scanf("%d", &num);
        node tmp = (node){num, i};
        if(i) {
            rt = q.lower_bound(tmp);
            lt = rt;
            lt--;
            if(rt == q.end()) {
                fq[i] = lt->x;
                dep[i] = dep[lt->id]+1;
            }
            else if(rt == q.begin()) {
                fq[i] = rt->x;
                dep[i] = dep[rt->id]+1;
            }
            else {
                if(dep[rt->id] > dep[lt->id]) {
                    fq[i] = rt->x;
                    dep[i] = dep[rt->id]+1;
                }
                else {
                    fq[i] = lt->x;
                    dep[i] = dep[lt->id]+1;
                }
            }
        }
        q.insert(tmp);
    }
    for(int i = 1; i < n; i++)
    {
        printf("%d", fq[i]);
        if(i == n-1) printf("\n");
        else printf(" ");
    }
    return 0;
}