主席树

简述

• 主席树是记录历史状态的权值线段树，实现方式是动态开点，节点重用。
• 数据值域和数量差距大时需要离散化。
• 需要初始化
• 动态开点才能以0开始作为值域

初始化

void init(int n) {
    for(int i = 0; i <= n; i++) {
        root[i] = 0;
    }
    for(int i = 0; i <= cnt; i++) {
        tree[i].l = tree[i].r = tree[i].sum = 0;
    }
    cnt = 0;
}

建立

• 对于第 i 次插入的数，重新建立一个新节点表示[1~i ]次插入的权值线段树（权值线段树：a[ i ] = t，即数值为i的有t个，维护出现的数的值在区间[l, r]的个数）
• 运用前缀和思想，因为个数是满足前缀和的相减性质的
• 模板：

struct node {
    int l, r, sum;
} tree[maxn*40];
int root[maxn], cnt;//root 是插入第i个数后新的权值线段树的根的==标号==
//在值域为[l, r]的主席树上插入一个数pos
void update(int l, int r, int &now, int lst, int pos) {
    tree[++cnt] = tree[lst], tree[cnt].sum++, now = cnt;
    if(l == r)
        return;
    int mid = (l + r) / 2;
    if(pos <= mid)
        update(l, mid, tree[now].l, tree[lst].l, pos);//调用tree[now].l时，下一步自动给左子树分配结点编号=++cnt
    else
        update(mid + 1, r, tree[now].r, tree[lst].r, pos);
}

int main() {
	init(n);
    for(int i = 1; i <= n; i++) {
        update(1, n, root[i], root[i-1], a[i]);
    }
	return 0;
}

查询

• 边二分边查询
• 主席树仅是查找工具，它维护的是原数组值域在[l, r]的数的个数，每加一个数就新建一个线段树的根，并通过结点重用延续前一个线段树的信息，并新添加当前插入的数的信息
• 左右子树表示的区间仅是为了区分数值大小
• 同时下到左/右子树是为了同步区间的大小：root[i] [l, r] 的个数有a个，root[j] [l, r]的个数有b个（j > i），则值在[l, r]区间内的新增的数的个数为 b - a ，若b - a >= k，则要查询的 (lf, rt) (指第lf次插入到第rt次插入)内的第k大在[l, r] 内，继续下放查找
• 模板：

//查询[x, y]区间第k大
int query(int l, int r, int x, int y, int k) {
    if(l == r)
        return l;
    int mid = (l + r) / 2;
    int sum = tree[tree[y].l].sum - tree[tree[x].l].sum;
    if(sum >= k)
        return query(l, mid, tree[x].l, tree[y].l, k);
    else
        return query(mid + 1, r, tree[x].r, tree[y].r, k - sum);
}
//main里
int ans = query(1, n, root[qul-1], root[qur], k);