【模板】主席树

发表于

经典的不加修改的询问第k小问题

同样是线段树 但是建树方法不同
将序列排序去重以后 节点值为每个数出现次数
查第k小就像平衡树那样查就行
利用前缀和进行相减
每加一个值只用添加修改的logn个节点
其他的指向上一棵树的同样节点就行
再次引用大佬的文章

CODE

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m,len,top,ans;
int a[210101],doe[210101],T[210101],L[4020202],R[4020202],s[4020202];
inline void read(int &x)
{
    int data=0,w=1;
    char ch=0;
    while (ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
    if (ch=='-') w=-1,ch=getchar();
    while (ch>='0'&&ch<='9') data=(data<<3)+(data<<1)+(ch^'0'),ch=getchar();
    x=data*w;
}
int build(int pre,int l,int r,int x)
{
	int now=++top;
	L[now]=L[pre];
	R[now]=R[pre];
	s[now]=s[pre]+1;
	if (l==r) return now;
	int mid=(l+r)/2;
	if (x<=mid) L[now]=build(L[pre],l,mid,x);
	if (x>mid) R[now]=build(R[pre],mid+1,r,x);
	return now;
}
void find(int pre,int now,int l,int r,int k)
{
	if (l==r)
	{
		ans=doe[l];
		return;
	}
	int zuo=s[L[now]]-s[L[pre]];
	int mid=(l+r)/2;
	if (k<=zuo) find(L[pre],L[now],l,mid,k);
	if (k>zuo) find(R[pre],R[now],mid+1,r,k-zuo);
}
int main()
{
	read(n); read(m);
	for (int i=1;i<=n;++i)
		{
			read(a[i]);
			doe[i]=a[i];
		}
	sort(doe+1,doe+1+n);
	len=unique(doe+1,doe+1+n)-1-doe;
	
	int x;
	for (int i=1;i<=n;++i)
	{
		x=lower_bound(doe+1,doe+1+len,a[i])-doe;
		T[i]=build(T[i-1],1,len,x);
	}
	int l,r,k;
	for (int i=1;i<=m;++i)
	{
		read(l); read(r); read(k);
		find(T[l-1],T[r],1,len,k);
		printf("%d\n",ans);
	}
	return 0;
}
--------------------------