主席树静态查找区间第k小模板(并没有杰少详细)即讲解

#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#define maxn 400100

using namespace std;

struct Data
{
    int val,id;
};

struct Tree
{
    int l,r,sum;
};

Data a[maxn];
Tree t[maxn*100];

int Rank[maxn],Root[maxn],cntt;

bool cmp(Data a,Data b)
{
    return a.val<b.val;
}

void init()
{
    cntt=1;
    Root[0]=0;
    t[0].l=t[0].r=t[0].sum=0;
    return;
}

void update(int val,int&root,int l,int r)
{
    t[cntt]=t[root];
    t[cntt].sum++;
    root=cntt;
    cntt++;

    if (l==r) return;

    int mid=(l+r)/2;
    if (val<=mid) update(val,t[root].l,l,mid);
    else update(val,t[root].r,mid+1,r);
    return;
}

int query(int rl,int rr,int k,int l,int r)
{
    if (l==r) return l;

    int sum=t[t[rr].l].sum-t[t[rl].l].sum;
    int mid=(l+r)/2;

    if (k<=sum) return query(t[rl].l,t[rr].l,k,l,mid);
    else return query(t[rl].r,t[rr].r,k-sum,mid+1,r);
}

int findx(int rl,int rr,int x,int l,int r)
{
    if (l==r) return 1;

    int sum=t[t[rr].l].sum-t[t[rl].l].sum;
    int mid=(l+r)/2;

    if (x<=mid) return findx(t[rl].l,t[rr].l,x,l,mid);
    else return sum+findx(t[rl].r,t[rr].r,x,mid+1,r);
}

int trans(int x,int n)
{
    a[n+1].val=0x3f3f3f3f;
    int l=1,r=n+1;
    while(l<r)
    {
        int mid=(l+r)/2;
        if (a[mid].val<=x) l=mid+1;
        else r=mid;
    }
    return l-1;
}

int main()
{
    int n,m;
    while(~scanf("%d %d",&n,&m))
    {
        for (int k=1;k<=n;k++)
            {scanf("%d",&a[k].val);a[k].id=k;}

        sort(a+1,a+n+1,cmp);

        for (int k=1;k<=n;k++)
            Rank[a[k].id]=k;

        init();

        for (int k=1;k<=n;k++)
        {
            Root[k]=Root[k-1];
            update(Rank[k],Root[k],1,n);
        }

        for (int k=1;k<=m;k++)
        {
            int l,r,kk;
            scanf("%d %d %d",&l,&r,&kk);
            printf("%d\n",a[query(Root[l-1],Root[r],kk,1,n)].val);
        }
    }
    return 0;
}


具体介绍请参见杰少的博客

这里说下杰少没介绍到的几个地方

我们知道update的时候是加一条链,其实对于每一个update,我都新建了一颗线段树,他的值的范围是1到n 

只是有些左子树或者右子树可以共用前面的而已,所以每个初始化的时候Root[k]=Root[k-1]

然后我们修改就行了

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值