题意就是给定一个无序数列,然后给定一堆询问,求区间[ s , t ]内的第k大的数。
查了很多资料,方法很多,看了半天只看懂了主席树怎么做,其余的做法以后补充。
主席树(函数式线段树):
首先是离散化,将所有数按升序映射到正整数。
用线段树统计这些正整数的出现次数(Sum数组),这样就可以快速找到第k大的数。
开始时,是一颗空树,所有的Sum都为0.
用T[n]表示用数列前n个数按上述方法建立的线段树的根(以后用根来指代整棵树)。
然后要求[S,T]区间的按值建立的线段树,就只需要把两颗线段树T[R]和T[S-1]相减(因为叶节点表示的是数出现的次数,满足区间减法)。
永久化数据结构思想:
永久化数据结构思想就是在修改的时候,不修改原来数据结构,而是新建节点,并且最大限度的复用原来的数据结构。
由于T[n]和T[n-1]之间结构一样并且,只有一个叶节点的值不同,所以最多有log2 (n) 个节点不同,于是相同的节点可以复用。
就是T[n]只需要多用改变的节点的空间,其余的直接使用前一棵树的节点,这样做的好处是保存了每个时间的线段树状态,也就是保存了整个操作历史。
由于没有破坏原来的数据结构,可以访问任意时刻的线段树(从根开始递归)。要做到这点,就没办法再用原来线段树的堆式存储了,而是增加了左右子树的指针。
也可以仿照SBT的写法直接用数组表示节点,用L[]和R[]表示左右子树的下标。
建树与查询:
求T[n]的时候就是在T[n-1]中递归,然后把所有经过的节点(也就是修改过的节点)都用新节点存起来。
最后在递归访问线段树的时候,同步访问R 和 S-1两颗线段树,用Sum[R]-Sum[S-1]来表示[L,R]区间的Sum值
就可以找到第K大的数了。
还有,既然一开始是一颗空树,就不需要建立一整棵颗空树了,直接用0表示空树,省去了开始的一棵树的空间,
这样的话需要的空间就是n*log2n
所以主席树的初始化变得跟SBT一样简洁:
L[0]=R[0]=Sum[0]=T[0]=TP=0;
代码:
主席树代码:(811MS 27552KB)
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <map>
#define maxn 100007
#define maxnn 1860007
using namespace std;
//主席树
int L[maxnn],R[maxnn],Sum[maxnn],T[maxn],TP;//左右子树,总和,树根,指针
void Add(int &rt,int l,int r,int x){//建立新树
++TP;L[TP]=L[rt];R[TP]=R[rt];Sum[TP]=Sum[rt]+1;rt=TP;//复制&新建
if(l==r) return;
int m=(l+r)>>1;
if(x <= m) Add(L[rt],l,m,x);
else Add(R[rt],m+1,r,x);
}
int Search(int TL,int TR,int l,int r,int k){//区间查询第k大
if(l==r) return l;//返回第k大的下标
int m=(l+r)>>1;
if(Sum[L[TR]]-Sum[L[TL]]>=k) return Search(L[TL],L[TR],l,m,k);
else return Search(R[TL],R[TR],m+1,r,k-Sum[L[TR]]+Sum[L[TL]]);
}
//常规
int n,m;
int value[maxn];
//离散化
struct A{
int x,id;
bool operator<(const A&B)const{return x<B.x;}
}ID[maxn];
map<int,int> MP;
int Rank[maxn];
int main(void)
{
int Test;scanf("%d",&Test);
while(Test-->0){
//读取输入+离散化
scanf("%d%d",&n,&m);
for(int i=1;i<=n;++i) {
scanf("%d",&value[i]);
ID[i].x=value[i];
ID[i].id=i;
}
sort(ID+1,ID+n+1);MP.clear();
for(int i=1;i<=n;++i){
Rank[i]=ID[i].x;
MP[ID[i].x]=i;
}
//初始化主席树
L[0]=R[0]=Sum[0]=T[0]=TP=0;
//建主席树
for(int i=1;i<=n;++i) Add(T[i]=T[i-1],1,n,MP[value[i]]);
//开始计算
for(int i=0;i<m;++i){
int s,t,k;scanf("%d%d%d",&s,&t,&k);
printf("%d\n",Rank[Search(T[s-1],T[t],1,n,k)]);
}
}
return 0;
}