主席树入门:
静态(只有查询,没有修改):
引入:为什么要引入主席树呢,我们先从一道题进行引入。这样的:就是给了一个序列,我要进行若干次的查询,就是给定区间[L,R],查询这个区间中第k大的数字。显然,多次查询就不大好办了,显然,我们第一想到的就是能否利用前缀和的思想。
那么我们就开门见山,原序列是a【】,假设a序列离散化完了,然后让rt【o】这棵树,代表当前插入了a[1]-a[o]这些数字后的这样的一颗线段树。我们建立n棵线段树。
那么想一下,如果让sum【节点号】代表这个节点代表区间的插入数字的个数,(就是类似于线段树中维护的那个值),假设我要求【L,R】区间里的第k小的,那么我们此时就想啊,我们建立的第R棵线段树,和第L棵线段树,对应位置上的数字一减是不是就是我在【L,R】中的所有数字哇,然后就是正常的查询第k个小的就好了。
但是问题来了,正常建立这么多线段树内存肯定是要炸的。于是有人就画了画就发现了,咦,我的第 i 颗线段树比第 i-1 颗线段树不就只增加了一个点而已嘛,就是一条链上的节点,那么影响的节点就是logn个。不变的节点就不让他变了,节点重复利用嘛!于是就出现了好多博客里的那种复杂的树。(其实并不复杂,但博主当时看了好一会儿)。
比如说(以区间第k大为例hdu2665题目戳这里http://acm.hdu.edu.cn/showproblem.php?pid=2665):
设n = 4,q= 1;
4个数分别为4, 1, 3 ,2;
ql = 1, qr = 3, k = 2;
emmm,盗完图了。
图的来源:
https://www.cnblogs.com/zyf0163/p/4749042.html
看不懂就数节点的编号。会发现这个节点改变的话就是会多加一个节点的。我们怎么写呢?感觉不会写欸:
void update(int& o, int l, int r, int last, int p){
o = ++ tot;
ls[o] = ls[last];
rs[o] = rs[last];
sum[o] = sum[last] + 1;
if(l == r) return;
int m = (l + r) >> 1;
if(p <= m) update(ls[o], l, m, ls[last], p);
else update(rs[o], m + 1, r, rs[last], p);
}
看这个代码,就是用last代表第i-1棵树,o是当前的第i棵树,然后我们走的路径上的点肯定是新加的。而原来,共享的就不变就好了。
注意:&o的意思是取地址,从哪来的就改变哪个值。这样rt【o】实际上每一次都是在改变的。
代码:
#include<bits/stdc++.h>
using namespace std;
const int N = 100000 + 5;
int a[N], b[N], rt[N * 20], ls[N * 20], rs[N * 20], sum[N * 20];
int n, k, tot, sz, ql, qr, x, q, T;
void Build(int& o, int l, int r){
o = ++ tot;
sum[o] = 0;
if(l == r) return;
int m = (l + r) >> 1;
Build(ls[o], l, m);
Build(rs[o], m + 1, r);
}
void update(int& o, int l, int r, int last, int p){
o = ++ tot;
ls[o] = ls[last];
rs[o] = rs[last];
sum[o] = sum[last] + 1;
if(l == r) return;
int m = (l + r) >> 1;
if(p <= m) update(ls[o], l, m, ls[last], p);
else update(rs[o], m + 1, r, rs[last], p);
}
int query(int ss, int tt, int l, int r, int k){
if(l == r) return l;
int m = (l + r) >> 1;
int cnt = sum[ls[tt]] - sum[ls[ss]];
if(k <= cnt) return query(ls[ss], ls[tt], l, m, k);
else return query(rs[ss], rs[tt], m + 1, r, k - cnt);
}
void work(){
scanf("%d%d%d", &ql, &qr, &x);
int ans = query(rt[ql - 1], rt[qr], 1, sz, x);
printf("%d\n", b[ans]);
}
int main(){
scanf("%d", &T);
while(T--){
scanf("%d%d", &n, &q);
for(int i = 1; i <= n; i ++) scanf("%d", a + i), b[i] = a[i];
sort(b + 1, b + n + 1);
sz = unique(b + 1, b + n + 1) - (b + 1);
tot = 0;
Build(rt[0],1, sz);
//for(int i = 0; i <= 4 * n; i ++)printf("%d,rt = %d,ls = %d, rs = %d, sum = %d\n", i, rt[i], ls[i], rs[i], sum[i]);
for(int i = 1; i <= n; i ++)a[i] = lower_bound(b + 1, b + sz + 1, a[i]) - b;
for(int i = 1; i <= n; i ++)
{
/// cout<<rt[i]<<" &&& "<<rt[i-1]<<endl;
update(rt[i], 1, sz, rt[i - 1], a[i]);
}
///for(int i = 0; i <= 5 * n; i ++)printf("%d,rt = %d,ls = %d, rs = %d, sum = %d\n", i, rt[i], ls[i], rs[i], sum[i]);
while(q --)work();
}
return 0;
}