归并树 学习笔记

归并树

Problem

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

示例：区间第 $k$ 大值，link。

数据强制在线，$n,q\le 10^5$。

Algorithm description

Method

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

每个节点是其子节点代表的数组归并后的数组，这样能使任意一个节点有序。
也就是说，建树时，每个节点代表了 $a_l$ 至 $a_r$ 归并后的数组。

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

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

Code

以 POJ2104 （见上文）为例，可以二分答案再查找。

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;
}