[caioj1441][主席树]第K小的数

【问题描述】

给n(1<=n<=100000)个数字
a[1],a[2],……,a[n（0<=a[i<=1000000000）,m（1<=m<=100000）次询问l到r之间的第k小的值。

【输入文件】

第一行为n和m。 接下来一行输入n个数。 接下来m行，每行三个数l,r和k。

【输出文件】

m行，每行对应一个答案。

【输入样例】

7 3
1 5 2 6 3 7 4
2 5 3
4 4 1
1 7 3

【输出样例】

5
6
3

【题解】

主席树的板子
其实吧，我觉得就是和lct有一点点相似，在一棵线段树上加限制条件使得他们变成不一样的树
而主席树是什么呢？先介绍一点东西：线段树是一棵完全二叉树，他的节点是比较多的。那么节点建出来多了，自然就有一些是没有用的。对于一个点，在线段树中从根到他只有一条路径。那么主席树其实就是这样的一堆路径合成一棵树啦
对于建树，我们先每个点都拿出一条链来。就是在原线段树中从根到他的那条链
然后，将这条链，与这条链之前那条链合并起来。合并完之后就是树啦
对于查找x~y中第k大的数，我们就将1~x-1的树拿出来，和1~y的树拿出来。在线段树的查找之间减一减，就可以得到啦

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct trnode
{
    int lc,rc,c;
}tr[2110000];int len;
int Rank[110000],s[110000];
struct LSnode
{
    int y,p;
}wr[110000];int n,m;
bool cmp(LSnode n1,LSnode n2){return n1.p<n2.p;}
int rt[110000];
void add(int &u,int l,int r,int p)
{
    if(u==0)u=++len;
    tr[u].c++;
    if(l==r)return ;
    int mid=(l+r)/2;
    if(p<=mid)add(tr[u].lc,l,mid,p);
    else add(tr[u].rc,mid+1,r,p);
}
void merge(int &x,int y)
{
    if(y==0)return ;
    if(x==0){x=y;return ;}
    tr[x].c+=tr[y].c;
    merge(tr[x].lc,tr[y].lc);
    merge(tr[x].rc,tr[y].rc);
}
int findKth(int x,int y,int l,int r,int k)
{
    if(l==r)return wr[l].p;//由于这里主席树管理的是排名，那么我们就要return排名后的 
    int lcx=tr[x].lc,rcx=tr[x].rc,lcy=tr[y].lc,rcy=tr[y].rc;
    int c=tr[lcx].c-tr[lcy].c;
    int mid=(l+r)/2;
    if(k<=c)return findKth(lcx,lcy,l,mid,k);
    else return findKth(rcx,rcy,mid+1,r,k-c);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&s[i]);
        wr[i].p=s[i];wr[i].y=i;
    }
    sort(wr+1,wr+1+n,cmp);
    for(int i=1;i<=n;i++)Rank[wr[i].y]=i;
    len=0;memset(rt,0,sizeof(rt));
    for(int i=1;i<=n;i++)
    {
        add(rt[i],1,n,Rank[i]);
        merge(rt[i],rt[i-1]);
    }
    while(m--)
    {
        int x,y,k;
        scanf("%d%d%d",&x,&y,&k);
        printf("%d\n",findKth(rt[y],rt[x-1],1,n,k));
    }
    return 0;
}