【超好懂的数据结构】可持久化线段树/主席树 详解~(update:11/12)

---

title : 主席树
date : 2021-8-18
tags : 数据结构,ACM
author : Linno

---

主席树

学名为可持久化线段树，可以用来解决线段树存储历史状态的问题。

我们在进行单点修改后，线段树只有logn个(一条链)的节点被修改，我们可以让修改后的树与修改前的树共享节点，节省时间和空间。

在学习主席树之前，我们先引入三个前置知识：离散化、动态开点，权值线段树。

离散化

对于较大的数据范围，只要将关键点记录下来，记录下rank，就能把数据缩小到可以接受的范围，以便建立线段树或其他数据结构来解决问题。

具体步骤

（1）将所有端点加入辅助数组；
（2）按坐标从小到大排序；
（3）去重；
（4）数据离散化，用hs数组记录端点的排名而非具体数字

vector<int>vt;
for(int i=1;i<=n;i++){
    cin>>a[i];
    vt.push(a[i]); //加入辅助容器
}
sort(vt.begin(),vt.end());//排序
vt.erase(unique(vt.begin(),vt.end()),vt.end()); //去重
for(int i=0；i<vt.size();i++){
    hs[vt[i]]=++tot; //存储为rank
}

动态开点

动态开点线段树可以避免离散化。

如果权值线段树的值域较大，离散化比较麻烦，可以用动态开点的技巧。

省略了建树的步骤，而是在具体操作中加入结点。

权值线段树

线段树的叶子节点保存的是当前值的个数。

每个节点保存区间左右端点以及所在区间节点的个数。

由于值域范围通常较大，一般会配合离散化或动态开点等策略优化空间。

应用

查找一个区间的第k大的值

查询某个数的排名

查询整个数组的排序

查询前驱和后继

单点修改

void update(int node,int start,int end,int pos){
    if(start==end) tr[node]++;
    else{
        int mid=start+end>>1;
        if(pos<=mid) update(node<<1,start,mid,pos);
        else update(node<<1|1,mid+1,end,pos);
    }
}//tr[i]表示值为i的元素个数,pos是要查找的位置

查询区间中的数出现次数

int query(int node,int start,int end,int ql,int qr){
    if(start==ql&&end==qr) return tr[node];
    int mid=start+end>>1;
    if(qr<=mid) return query(node<<1,start,mid,ql,qr);
    else if(ql>mid) return query(node<<1|1,mid+1,end,ql,qr);
    else return query(node<<1,start,mid,ql,qr)+query(node<<1|1,mid+1,end,ql,qr);
}//对单点查询同样适用
<