码迷,mamicode.com
首页 > 其他好文 > 详细

主席树模板(区间第k小)

时间:2017-11-04 00:14:59      阅读:180      评论:0      收藏:0      [点我收藏+]

标签:insert   ios   div   pac   --   scanf   code   ++   names   

#include<cmath>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 200010
using namespace std;
int n,m,cnt;
struct Tree
{
    int l,r,sum;
}T[MAXN*60];
int rank[MAXN],root[MAXN];
struct data
{
    int val,ind;
    bool operator < (const data &a) const
    {
        return val<a.val;
    }
}a[MAXN];
void insert(int &num,int &x,int l,int r)
{
    T[cnt++]=T[x];
    x=cnt-1;
    T[x].sum++;
    if(l==r) return;
    int mid=(l+r)/2;
    if(num<=mid) insert(num,T[x].l,l,mid);
    else insert(num,T[x].r,mid+1,r);
}
int query(int i,int j,int k,int l,int r)
{
    if(l==r) return l;
    int t=T[T[j].l].sum-T[T[i].l].sum;
    int mid=(r+l)/2;
    if(k<=t) return query(T[i].l,T[j].l,k,l,mid);
    else return query(T[i].r,T[j].r,k-t,mid+1,r);
}
int main()
{
    T[0].l=T[0].r=T[0].sum=root[0]=0;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i].val);
        a[i].ind=i;
    }
    sort(a+1,a+n+1);
    for(int i=1;i<=n;i++)
    {
        rank[a[i].ind]=i;
    }
    cnt=1;
    for(int i=1;i<=n;i++)
    {
        root[i]=root[i-1];
        insert(rank[i],root[i],1,n);
    }
    while(m--)
    {
        int i,j,k;
        scanf("%d%d%d",&i,&j,&k);
        printf("%d\n",a[query(root[i-1],root[j],k,1,n)].val);
    }
}

 

主席树模板(区间第k小)

标签:insert   ios   div   pac   --   scanf   code   ++   names   

原文地址:http://www.cnblogs.com/wjxgy/p/7780603.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!