主席树poj2104

主席树:其实就是开了n个前缀线段树,但是每次只更新logn个节点信息,达到可以利用历史信息来求得所需答案
其最简单的应用就是区间第k大
以下是大致的建树过程
主席树poj2104_第1张图片

#include
#include
#include 
using namespace std;
const int N=1e5+10;
struct node
{
    int l,r,val;
}tree[N*20];
int root[N],cnt;
int a[N],b[N],aa[N];
void update(int x,int &rt,int l,int r)
{
    tree[++cnt]=tree[rt];
    rt=cnt;
    tree[rt].val++;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(x<=mid)
        update(x,tree[rt].l,l,mid);
    else
        update(x,tree[rt].r,mid+1,r);
}
int query(int L,int R,int k,int l,int r)
{
    if(l==r)return l;
    int tmp=tree[tree[R].l].val-tree[tree[L].l].val;
    int mid=(l+r)>>1;
    if(tmp>=k)return query(tree[L].l,tree[R].l,k,l,mid);
    else return query(tree[L].r,tree[R].r,k-tmp,mid+1,r);
}
int main()
{
   int n,m;
   cin>>n>>m;
   for(int i=1;i<=n;i++)
   {
        scanf("%d",&a[i]);
        b[i]=a[i];
   }
   sort(b+1,b+n+1);
   for(int i=1;i<=n;i++)
       aa[i]=lower_bound(b+1,b+n+1,a[i])-b;
   cnt=0;
   for(int i=1;i<=n;i++)
   {
       root[i]=root[i-1];
       update(aa[i],root[i],1,n);
   }
   while(m--)
   {
       int x,y,k;
       scanf("%d%d%d",&x,&y,&k);
       printf("%d\n",b[query(root[x-1],root[y],k,1,n)]);
   }
}

你可能感兴趣的:(主席树)