今天学习了主席树,我觉得我可以去黑入美国国防部的网络了。
所谓主席树,它的真正含义和主席两个字完全没有联系。
主席树的作用是保存历史版本的线段树。
不知为何我对这张图片情有独钟。
划圈的部分是单点修改后的新的线段树。
下面代码是一道主席树模板题,求区间第k小的数的值。
此题思路就是每次插入一个数,就新建一颗权值线段树,变成主席树。
输入x到y求第k小的数,就是用第y个版本的线段树减去第x-1个版本的线段树,这样所剩下的权值线段树就是x到y这一区间上的数组成的权值线段树了,然后查询第k小。
就很妙。
//#pragma GCC optimize(3,"Ofast","inline")
//#include<unordered_map>
#include<cstdio>
#include<iostream>
#include<cmath>
#include<functional>
#include<cstring>
#include<string>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
#include<stack>
#include<vector>
using namespace std;
typedef long long ll;
const int INF=0x3f3f3f;
int mod=1000000000;
const int maxn=2e5+7;
int n,m,cnt,root[maxn],a[maxn];
//n是几个点,m是几次查询,cnt就是个计数的,类似于是邻接表里的cnt。
//root数组储存每个版本线段树的根节点编号,类似于邻接表head数组,a数组储存输入的n个数。
struct node
{
int l,r,sum;
}t[maxn*40];//线段树的结构体
vector<int >v;//此物是stl的容器,在这里用于离散化。
int getid(int x)
{
return lower_bound(v.begin(),v.end(),x)-v.begin()+1;
}//这个函数是找出以排好序的容器中x在哪个位置,update的时候用
void update(int l,int r,int &now,int last,int id)
{
t[++cnt]=t[last];//新建一个点等于上一个版本的点
t[cnt].sum++;//修改sum
now=cnt;//带&修改上次递归的点的左或者右孩子
if(l==r) return ;
int mid=(l+r)>>1;
if(id<=mid)update(l,mid,t[now].l,t[last].l,id);//套路安排左边
else update(mid+1,r,t[now].r,t[last].r,id);//套路安排右边
}
int query(int l,int r,int x,int y,int k)//查询
{
if(l==r)return l;
int mid=(l+r)>>1;
int sum=t[t[y].l].sum-t[t[x].l].sum;//这里找出左孩子所代表的区间里有多少个元素
if(sum>=k) return query(l,mid,t[x].l,t[y].l,k);//如果k小于这个数,那么要找的数就在左边
else return query(mid+1,r,t[x].r,t[y].r,k-sum);//反之,找右边,那么在右边的话就是查询第k-sum小的数了
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
{
scanf("%d",&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++)
update(1,n,root[i],root[i-1],getid(a[i]));//插入
int l,r,q;
while(m--)//每次查找
{
scanf("%d%d%d",&l,&r,&q);
printf("%d\n",v[query(1,n,root[l-1],root[r],q)-1]);
}
return 0;
}