归并树
归并树是区域树的一种,它的每个节点都是一个数组,通常是按某一顺序将数组排好序,便于统计。
每次区间操作只会涉及到 O ( log n ) O(\log n) O(logn)个节点,排好序的数组又可以用二分查找来统计某个量,所以每次查询的复杂度通常为 O ( log 2 n ) O(\log^2n) O(log2n)。
int n, m, a[N], num[N];
struct tnode
{
vector<int> ver;
tnode* lson,* rson;
tnode(int l = 0, int r = -1)
{
ver.clear(); lson = rson = NULL;
for (int i = l; i <= r; ++ i) ver.push_back(a[i]);
}
void maintain()
{
ver.resize(lson->ver.size() + rson->ver.size()); // merge之前一定要resize !!!
merge(lson->ver.begin(), lson->ver.end(), rson->ver.begin(), rson->ver.end(), ver.begin());
}
};
tnode* tre;
tnode* newt(int l, int r)
{
if (l == r) return new tnode(l, l);
tnode* ret = new tnode();
int mid = (l + r) >> 1;
ret->lson = newt(l, mid); ret->rson = newt(mid + 1, r);
ret->maintain();
return ret;
}
int queryt(tnode* cur, int l, int r, int ql, int qr, int val)
{
if (ql <= l && r <= qr) return lower_bound(cur->ver.begin(), cur->ver.end(), val) - cur->ver.begin();
int mid = (l + r) >> 1, ret = 0;
if (ql <= mid) ret += queryt(cur->lson, l, mid, ql, qr, val);
if (qr > mid) ret += queryt(cur->rson, mid + 1, r, ql, qr, val);
return ret;
}