归并树 学习笔记

归并树

Problem

给定一个长为 n n n 的数组。
q q q 次查询,每次查询包括一个区间,需要回答关于区间元素排序的问题

示例 :区间第 k k k 大值,link

数据强制在线 , n , q ≤ 1 0 5 n,q \leq 10^5 n,q≤105。

Algorithm description

Method

归并树顾名思义,就是逐个归并的树。

每个节点是其子节点代表的数组归并后的数组,这样能使任意一个节点有序。

也就是说,建树时,每个节点代表了 a l a_l al 至 a r a_r ar 归并后的数组

我们在实现时,为了节约空间,可以开 dep × n \text{dep}\times n dep×n 的归并数组,每一层为下一层归并的结果。

这种树的优势在于,能够像线段树一样 O ( log ⁡ n ) O(\log n) O(logn) 统计出可以叠加的排序的结果

Code

以 POJ2104 (见上文)为例,可以二分答案再查找。

cpp 复制代码
int sorted[25][2000001], a[200001];

void build(int l, int r, int dep){
    if(l == r){
        sorted[dep][l] = a[l]; return;
    }
    int mid = l + r >> 1;
    build(l, mid, dep + 1);
    build(mid + 1, r, dep + 1);
    // Merging.
    int pos = l, pos1 = mid + 1, tot = l;
    while(pos <= mid && pos1 <= r){
        if(sorted[dep + 1][pos] < sorted[dep + 1][pos1])
            sorted[dep][tot] = sorted[dep + 1][pos], pos++, tot++;
        else
            sorted[dep][tot] = sorted[dep + 1][pos1], pos1++, tot++;
    }
    while(pos <= mid) sorted[dep][tot] = sorted[dep + 1][pos], pos++, tot++;
    while(pos1 <= r) sorted[dep][tot] = sorted[dep + 1][pos1], pos1++, tot++;
}
int query(int l, int r, int x, int y, int tar, int dep){
    if(x <= l && r <= y){
    	// Here it is possible to calculate the sum thanks to the merging above.
        return upper_bound(sorted[dep] + l, sorted[dep] + r + 1, tar) - sorted[dep] - l;
    }
    int mid = l + r >> 1, ans = 0;
    if(x <= mid) ans += query(l, mid, x, y, tar, dep + 1);
    if(mid < y) ans += query(mid + 1, r, x, y, tar, dep + 1);
    return ans;
}
相关推荐
CoovallyAIHub15 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
NAGNIP16 小时前
Serverless 架构下的大模型框架落地实践
算法·架构
moonlifesudo16 小时前
半开区间和开区间的两个二分模版
算法
moonlifesudo16 小时前
300:最长递增子序列
算法
CoovallyAIHub21 小时前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!
深度学习·算法·计算机视觉
CoovallyAIHub1 天前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
聚客AI2 天前
🙋‍♀️Transformer训练与推理全流程:从输入处理到输出生成
人工智能·算法·llm
大怪v2 天前
前端:人工智能?我也会啊!来个花活,😎😎😎“自动驾驶”整起!
前端·javascript·算法
惯导马工2 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
骑自行车的码农2 天前
【React用到的一些算法】游标和栈
算法·react.js