poj 2104

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_36651153/article/details/81487096

看着qscdalao的视频,初步学习了主席树

自己手动模拟了一遍,才理解线段树是怎么建的

代码如下

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=1e5+7;
int n,m,x,y,k,root[maxn],a[maxn],ha[maxn],hs,cnt;
struct Node
{
    int l,r,sum;
}T[maxn*40];
int getid(int x)
{
    return lower_bound(ha,ha+hs,x)-ha+1;
}
void Update(int l,int r,int &x,int y,int pos)
{
    T[++cnt]=T[y]; T[cnt].sum++; x=cnt;
    if(l==r) {
        return ;
    }
    int mid=l+r>>1;
    if(pos<=mid) {
        Update(l,mid,T[x].l,T[y].l,pos);
    }
    else{
        Update(mid+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 mid=l+r>>1;
    int sum=T[T[y].l].sum-T[T[x].l].sum;
    if(sum>=k) {
        Query(l,mid,T[x].l,T[y].l,k);
    }
    else{
        Query(mid+1,r,T[x].r,T[y].r,k-sum);
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) {
        scanf("%d",&a[i]);
        ha[i-1]=a[i];
    }
    sort(ha,ha+n);
    hs=unique(ha,ha+n)-ha;
    for(int i=1;i<=n;i++) {
        Update(1,n,root[i],root[i-1],getid(a[i]));
    }
    for(int i=1;i<=m;i++) {
        scanf("%d%d%d",&x,&y,&k);
        printf("%d\n",ha[Query(1,n,root[x-1],root[y],k)-1]);
    }
}

猜你喜欢

转载自blog.csdn.net/qq_36651153/article/details/81487096