显然网上已经有太多的人写过这道题的题解了,而且大家都写得挺详细的,这里我就分析一份代码,没学过的同学在看了别人的博客之后看这个代码分析,应该可以很快解决主席树的基础学习
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<map>
#include<queue>
#include<set>
using namespace std;
const int MAXN = 100010;
struct Node{
int L,R,sum;
};
Node T[MAXN*20];
int timer,root[MAXN],n,m,rank[MAXN],ll,rr,kk;
struct Line{
int x,idx;
}a[MAXN];
bool cmp(const Line& A,const Line& B){return A.x<B.x;}
void insert(int num,int &x,int l,int r){新的节点为x,之前只是给root赋了值,但是并没有给空间,所以我们下一句话就给空间
T[++timer]=T[x];x=timer;++T[x].sum;//这里的新的空间开给x这个
if(l==r) return;//如果l==r,那么已经更新到叶节点了,返回
int mid=(l+r)>>1;
if(num<=mid)insert(num,T[x].L,l,mid);//往作或者往右赋值,注意线段树存的是第几个数,而不是数
else insert(num,T[x].R,mid+1,r);
}
int query(int x,int y,int k,int l,int r){
if(l==r) return l;//返回位置
int mid=(l+r)>>1;
int t=T[T[y].L].sum-T[T[x].L].sum;//版本y的主席树减去版本x的主席树的左区间,也就是说,在y这个版本和x这个版本之间,左区间到底插入了多少数的意思
if(k<=t) return query(T[x].L,T[y].L,k,l,mid);//分别递归左右区间
else return query(T[x].R,T[y].R,k-t,mid+1,r);
}
int main(){
T[0].L=T[0].R=T[0].sum=0;
root[0]=0;//首先对建一棵0节点的空树
while(scanf("%d%d",&n,&m)!=EOF){
for(register int i=1;i<=n;i++){
scanf("%d",&a[i].x);
a[i].idx=i;//读入,idx是指i这一位的数组下标
}
sort(a+1,a+n+1,cmp);//按照x关键字从小到大排序
for(register int i=1;i<=n;i++) rank[a[i].idx]=i;
timer=1;//rank表示原来的第idx位,现在的标号是i
for(register int i=1;i<=n;i++){
root[i]=root[i-1];//先赋值一棵树过去
insert(rank[i],root[i],1,n);//然后再插入,插入的时候,用rank[i]插入,原因是按照最开始的插入顺序的排序之后的位置进行插入,相当于离散化了,真的是非常厉害,然后新构建root[i]这棵树
}
while(m--){
scanf("%d%d%d",&ll,&rr,&kk);
printf("%d\n",a[query(root[ll-1],root[rr],kk,1,n)].x); //函数式线段树支持两棵线段树的加减,让我们来看看query函数
}
}
return 0;
}
注:
代码来源于 http://www.cnblogs.com/oyking/p/3230296.html
我自己敲了一遍 OTZ oyking %%%