1. 程式人生 > >A - K-th Number POJ - 2104 -主席樹第一彈-第K大的數

A - K-th Number POJ - 2104 -主席樹第一彈-第K大的數

  • 感謝:http://www.cnblogs.com/zyf0163/p/4749042.html
  • https://blog.csdn.net/qq_24451605/article/details/49031123
  • 裸題多次查詢給定區間L—R內第K大的數
  • #include<iostream>
    #include<cstdio>
    #include<algorithm>
    using namespace std;
    #define maxn 123456
    struct node
    {
        int l,r,sum;
    } tree[maxn*30];
    int n,cnt,k,snum,a[maxn];
    int m,root[maxn],num[maxn];
    void build(int &u,int l,int r)
    {
        u=++cnt;
        tree[u].sum=0;
        if(l==r)return;
        int mid=(l+r)/2;
        build(tree[u].l,l,mid);
        build(tree[u].r,mid+1,r);
    }
    void updata(int p,int &u,int l,int r,int x)
    {
        u=++cnt;
        tree[u]=tree[p];
        tree[u].sum++;
        int mid=(l+r)/2;
        if(l==r)return;
        if(x>mid)
            updata(tree[p].r,tree[u].r,mid+1,r,x);
        else
            updata(tree[p].l,tree[u].l,l,mid,x);
    }
    int query(int t1,int t2,int l,int r,int k)
    {
        if(l==r)return l;
        int mid=(l+r)/2;
        int lf=tree[tree[t2].l].sum-tree[tree[t1].l].sum;
        if(lf>=k)
            return query(tree[t1].l,tree[t2].l,l,mid,k);
        else
            return query(tree[t1].r,tree[t2].r,mid+1,r,k-lf);
    }
    int main()
    {
        while(~scanf("%d%d",&n,&m))
        {
            cnt=0;
            for(int i=1; i<=n; i++)
            {
                scanf("%d",&a[i]);
                num[i]=a[i];
            }
            sort(num+1,num+n+1);
            snum=unique(num+1,num+1+n)-num-1;
            build(root[0],1,snum);
            for(int i=1; i<=n; i++)
            {
                int x=lower_bound(num+1,num+1+n,a[i])-num;
                updata(root[i-1],root[i],1,snum,x);
            }
            while(m--)
            {
                int l,r,k;
                scanf("%d%d%d",&l,&r,&k);
                int x=query(root[l-1],root[r],1,snum,k);
                printf("%d\n",num[x]);
            }
        }
        return 0;
    }