主席樹入門+模板 POJ 2104

查詢區間第K大,而且沒有修改。

主席樹的原理就是在現有的一顆線段樹上不斷加入新的節點,而加入的對於現有線段樹的影響另開一條鏈記錄下來,這樣我們就可以保存線段樹的歷史版本,在查詢時只需要把區間減一下就可以了。

#include <vector>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 1e5+7;
vector<int>q;
int a[maxn],root[maxn],cnt=0;
struct node{int l,r,sum;}T[maxn*20];
int getid(int x){return lower_bound(q.begin(),q.end(),x)-q.begin()+1;}
void update(int l,int r,int x,int &y,int pos)
{
    T[++cnt]=T[x];T[cnt].sum++;y=cnt; //開闢新的鏈
    if(l==r) return ;
    int m = (l+r)>>1;
    if(m>=pos) update(l,m,T[x].l,T[y].l,pos);//修改新開闢的鏈
    else update(m+1,r,T[x].r,T[y].r,pos);
}
int query(int l,int r,int x,int y,int k)
{
    if(l==r) return l;
    int m = (l+r)>>1;
    int sum = T[T[y].l].sum-T[T[x].l].sum;
    if(sum>=k) query(l,m,T[x].l,T[y].l,k);
    else query(m+1,r,T[x].r,T[y].r,k-sum);
}
int main()
{
    int i,n,m,k,x,y;
    scanf("%d%d",&n,&m);
    for(i=1;i<=n;i++) scanf("%d",a+i),q.push_back(a[i]);
    sort(q.begin(),q.end());q.erase(unique(q.begin(),q.end()),q.end());
    for(i=1;i<=n;i++)
        update(1,n,root[i-1],root[i],getid(a[i]));
    for(i=1;i<=m;i++)
    {
        scanf("%d%d%d",&x,&y,&k);
        printf("%d\n",q[query(1,n,root[x-1],root[y],k)-1]);
    }
    return 0;
}


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章