首页 > 代码库 > AC日记——可怜的狗狗 洛谷 P1533

AC日记——可怜的狗狗 洛谷 P1533

可怜的狗狗

 

思路:

  主席树第k大;

 

代码:

#include <bits/stdc++.h>using namespace std;#define maxn 300001#define maxn_ maxn*30int n,m,lc[maxn_],rc[maxn_],dis[maxn_],root[maxn];int tot,ai[maxn],bi[maxn],size;inline void in(int &now){    char Cget=getchar();now=0;    while(Cget>9||Cget<0) Cget=getchar();    while(Cget>=0&&Cget<=9)    {        now=now*10+Cget-0;        Cget=getchar();    }}void build(int &now,int l,int r){    now=++tot;    if(l==r)return;    int mid=l+r>>1;    build(lc[now],l,mid);    build(rc[now],mid+1,r);}void add(int &now,int pre,int l,int r,int to){    now=++tot,dis[now]=dis[pre]+1;    if(l==r) return;int mid=l+r>>1;    if(to<=mid) add(lc[now],lc[pre],l,mid,to),rc[now]=rc[pre];    else add(rc[now],rc[pre],mid+1,r,to),lc[now]=lc[pre];}int query(int now,int pre,int l,int r,int k){     if(l==r) return l;int mid=l+r>>1;    int di=dis[lc[now]]-dis[lc[pre]];    if(k<=di) return query(lc[now],lc[pre],l,mid,k);    else return query(rc[now],rc[pre],mid+1,r,k-di);}int main(){    in(n),in(m);    for(int i=1;i<=n;i++) in(ai[i]),bi[i]=ai[i];    sort(bi+1,bi+n+1),size=unique(bi+1,bi+n+1)-bi-1;    build(root[0],1,size);    for(int i=1;i<=n;i++)    {        int to=lower_bound(bi+1,bi+size+1,ai[i])-bi;        add(root[i],root[i-1],1,size,to);    }    int l,r,k;    while(m--)    {        in(l),in(r),in(k);        printf("%d\n",bi[query(root[r],root[l-1],1,size,k)]);        }    return 0;}

 

AC日记——可怜的狗狗 洛谷 P1533