主席树&&蒟蒻的学习

主席树是一种可持久化的数据结构——可持久化线段树(因为其具有函数性质也叫函数式线段树)。

朴素做法:对于每一个版本都新建一个线段树,空间复杂度会爆炸

但是可以发现,两版本之间有很多的相同之处,如果能利用上这些相同的不饿妇女,那么就可以节省很大一部分空间.

为了利用版本之间相同的内容,我们可以在构建新版本的线段树的时候,把没有变动的部分连回上一个版本的那个部分上,这样就可以实现0浪费

 

 解:

对于原序列的每一个前缀[1,2 ……, i]建立出一棵线段树维护值域上每个数出现的次数,则其树是可减的。

先看权值线段树:

权值线段树就是对一个值域上值的个数进行维护的线段树。

 问题:对于一段数字序列,每次询问其第k小值

先建树然后对于每次询问递归查找即可。要找的数如果在左子树上,就递归查找左子树上的第k大;如果在右子树上,那么就递归查找右子树上的第(k - 左子树值)大

有了权值线段树的基础,然后可以看题目了

首先我们需要对数据进行离散化处理。因为权值线段树维护的是值域上值的个数,数据范围很大,显然开不出那么大的线段树。因为问题只与数的大小相关,而与数本身是多少无关,所以我们可以对数进行离散化,只保存其大小关系即可。

然后用主席树构造一棵可持久化权值线段树,对于每个数字,将其离散化,新建一个版本的的权值线段树,然后插入这个离散化后的数字。

对于查询操作L,R,我们利用主席树的函数性质,用R那个版本的权值线段树减去L-1那个版本的权值线段树,在得到的权值线段树中查找第k小值就可以了,类似前缀和。

这样为什么正确?因为权值线段树储存的是值域上值的个数,我么用R的版本的去权值线段树减去L-1版本的权值线段树,得到的就是维护[L, R]上值的个数的权值线段树。

 问题:如何存储主席树?

首先,我们可以建立一个数组int root[maxn];来存储各个根节点的编号。对于儿子结点,可以看出主席树不像线段树可以用当前结点编号乘2和乘2加1来得到左右儿子的结点编号。于是我们可以这样做:用一个struct储存当前节点的左右儿子节点的编号和当前节点的值,然后用这个struct开一个内存池,每新建一个主席树的节点,就从内存池里取一块新的空间送给这个结点。取空间从hjt[1]开始,hjt[0]充当NULL

问题:用不用像线段树那样先构建好整个主席树然后再执行操作?

不用,准确的说程序已经帮我们构建好了。因为一开始树里什么都没有,所以所有的节点的l,r,sum都是0,而全局变量和数组都是默认赋值为0的。所以我们直接边插入边建树就ok了

问题:如何进行两树之间的减法?用不用新建一个权值线段树令其等于主席树两版本之差然后对这个权值线段树进行询问?

不用,可以在询问的时候便递归边减

(借鉴了AgOH大佬b站视频)

代码

#include<bits/stdc++.h>
using namespace std;
const int maxn = 2e5 + 5;
int a[maxn];
vector<int> v;
inline int getid(int x){return lower_bound(v.begin(), v.end(), x) - v.begin() + 1;}
struct node{
    int l, r, sum;
}hjt[maxn * 40];
int cnt, root[maxn];
void insert(int l, int r, int pre, int &now, int p){
    hjt[++cnt] = hjt[pre];
    now = cnt;
    hjt[now].sum++;
    if(l == r) return;
    int m = (l + r) >> 1;
    if(p <= m) insert(l, m, hjt[pre].l, hjt[now].l, p);
    else insert(m + 1, r, hjt[pre].r, hjt[now].r, p);
}
int query(int l, int r,int L, int R, int k){
    if(l == r) return l;
    int m = (l + r) >> 1;
    int tmp = hjt[hjt[R].l].sum - hjt[hjt[L].l].sum;
    if(k <= tmp) return query(l, m, hjt[L].l, hjt[R].l, k);
    else return query(m + 1, r, hjt[L].r, hjt[R].r, k - tmp);
}
int main(){
    int n, m;
    cin >> n >> m;
    for(int i = 1; i <= n; i++){
        cin >> a[i];
        v.push_back(a[i]);
    }
    sort(v.begin(), v.end());
    v.erase(unique(v.begin(), v.end()), v.end());
    for(int i = 1; i <= n; i++){
        insert(1, n, root[i - 1], root[i], getid(a[i]));
    }
    while(m--){
        int l, r, k;
        cin >> l >> r >> k;
        cout << v[query(1, n, root[l - 1], root[r], k) - 1] << "\n";
    }
}

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值