• 静态主席树总结(静态区间的k大)


    静态主席树总结(静态区间的k大)

    首先我们先来看一道题

    给定N个正整数构成的序列,将对于指定的闭区间查询其区间内的第K小值。       输入格式:
    第一行包含两个正整数N、M,分别表示序列的长度和查询的个数。
    第二行包含N个正整数,表示这个序列各项的数字。
    接下来M行每行包含三个整数 l, r, kl,r,k , 表示查询区间[l, r][l,r]         内的第k小值。
    输出格式:
    输出包含k行,每行1个正整数,依次表示每一次查询的结果
    

    对于100%的数据满足:(1 leq N, M leq 2cdot 10^5) (1≤N,M≤2⋅10^5)
    对于数列中的所有数(a_i),都有(-10^9leq a_ileq 10^9)

    基本思路

    这个题目看上去很像一道线段树或者树状数组之类的裸题,但是仔细想想,区间第(k)小是线段树等数据结构维护不了的,这个时候,我们就需要引进一种新的数据结构,就是可持久化线段树,也就是主席树。(可持久化数据结构是可以访问历史版本的,这里也可以做到,但是这里不需要访问历史版本,我们只利用可持久化数据结构的思想)
    主席树的本质上是N颗值域线段树(不知道值域线段树的请自行转走),对于每一颗线段树我们都维护从序列开始到这个元素的值域(即第(i)颗线段树维护的区间是第一号元素到第(i)号元素的值域)。
    但实际上我们没有那么多时间和空间去维护(n)颗线段树,所以我们就要想,每一颗线段树由于值域相同,它们的形状是完全相同的,并且对于一颗第(i)颗线段树来说,它相对于第(i-1)颗线段树只增加了一个值,放在值域中,也就只有包含这个值的log个节点不同,所以对于每一颗线段树,我们只需要新开(log)个节点,其它的节点就用第(i-1)颗线段树的节点(如果你会可持久化数组,就会发现这其实跟可持久化数组很像,准确的说可持续化数组的模型就是主席树)。

    注意要离散化
    实现过程

    构建

    我们先开一个root数组来保存每一颗线段树的根,对于每一个线段树的节点记录它的值,左儿子和右儿子的编号,在构建第i颗线段树时,我们要同时访问第i-1颗线段树,每次构建一个节点之后,对于不包含这个新增值的儿子我们就直接将第i-1颗线段树的相应的那个儿子作为第i颗线段树的这个儿子。

    比如说,假设离散化之后的值域是15,第i号元素是1,我们先构建根节点,然后发现这个节点左儿子的值域是12,右儿子的值域是3~5,右儿子的值域不包括1,所以右儿子就直接用第i-1颗线段树的右儿子,而此时我们就新建一个节点作为这个节点的左儿子,值为第i-1颗线段树的左儿子的值+1。

        int modify(int l,int r,int x,int k)
        {
            //x表示上一颗线段树当前节点的标号
            //k表示需要新增的元素
            int y=++cnt;//新建当前节点,y位编号
            t[y]=t[x];//将上一颗线段树的节点的信息传递给当前节点
            t[y].x++;/*因为不包含k的节点不会被访问,所以实质上只要被访问过的节点都要加1*/
            if(l==r)return y;
            int mid=(l+r)>>1;
            if(k<=mid)t[y].l=modify(l,mid,t[x].l,k);
            else t[y].r=modify(mid+1,r,t[x].r,k);/*根据k值修改左右儿子信息*/
            return y;//将当前节点的编号号返回上一层
        }
    

    查询

    主席树的查询跟值域线段树的查询差不多,值域线段树的查询大家都会吧,我这里就不再赘述,不过,主席树每次需要同时查询两颗线段树,如果我们需要查询([l,r])闭区间中第(k)小的值,我们就查询第(l-1)颗线段树和第(r)颗线段树的信息,由于所有线段树维护的值域完全一样,所以我们可以用第r颗线段树询问到的值减去第(l-1)颗线段树的值,就可以得出([l,r])闭区间的值。(注意:你查询到的是离散之后的值,你需要输出的是离散之前的值)

    具体实现过程

        int query(int l,int r,int la,int no,int k)
        {
            //la,no,分别表示你要查询的两颗线段树的相应节点编号
            if(l==r)return l;/*如果节点内只有一个值,这就是第k大,直接返回*/
            int l1=t[la].l,l2=t[no].l,r1=t[la].r,r2=t[no].r;
            //l1,r1,l2,r2分别表示这两个节点的左右儿子。
            int s=t[l2].s-t[l1].s  ,  mid=(l+r)>>1;
            if(s>=k)return query(l,mid,l1,l2,k);
            else return query(mid+1,r,r1,r2,k-s);
        }
    

    代码

        #include<bits/stdc++.h>
        using namespace std;
        inline int gi()
        {
            char a=getchar();int b=0;
                while(a<'0'||a>'9')a=getchar();
            while(a>='0'&&a<='9')b=b*10+a-'0',a=getchar();
            return b;
        }
        const int N=1e6+20;
        struct ljq
        {
            int x,id;
        }b[N];
        struct tree
        {
           int l,r,s;
        }t[N*5];
        int cmp(ljq x,ljq y){return x.x<y.x;}
        int a[N],p[N],root[N],n,m,cnt;
        void work1()
        {
            n=gi();m=gi();
            for(int i=1;i<=n;++i)
                b[i].x=gi(),b[i].id=i;
            sort(b+1,b+n+1,cmp);
            b[0].x=-2e9;
            for(int s=0,i=1;i<=n;++i)
            {
                if(b[i].x!=b[i-1].x)p[++s]=b[i].x;
                a[b[i].id]=s;
            }
        }
        void bt(int l,int r,int x)
        {
            if(l==r)return;
            int mid=(l+r)>>1;
            t[x].l=++cnt;
            t[x].r=++cnt;
            bt(l,mid,t[x].l);
            bt(mid+1,r,t[x].r);
        }
        void work2(int l,int r,int la,int no,int x)
        {
            t[no].s=t[la].s+1;
            if(l==r)return;
            int mid=(l+r)>>1;
            t[no].l=t[la].l;
            t[no].r=t[la].r;
            if(x<=mid)
            {
                t[no].l=++cnt;
                work2(l,mid,t[la].l,t[no].l,x);
            }
            else
            {
                t[no].r=++cnt;
                work2(mid+1,r,t[la].r,t[no].r,x);
            }
        }/*这个构建主席树的实现过程和上面略有不同,上面的更方便,是我在打带修改的主席树的时候写的,这里我懒得改了,仅做参考*/
        int query(int l,int r,int la,int no,int k)
        {
            if(l==r)return l;
            int l1=t[la].l,l2=t[no].l,r1=t[la].r,r2=t[no].r;
            int s=t[l2].s-t[l1].s,mid=(l+r)>>1;
            if(s>=k)return query(l,mid,l1,l2,k);
            else return query(mid+1,r,r1,r2,k-s);
        }
        int main()
        {
            work1();
            root[0]=++cnt;
            bt(1,n,1);
            for(int i=1;i<=n;++i)
            {
                root[i]=++cnt;
                work2(1,n,root[i-1],root[i],a[i]);
            }
            while(m--)
            {
                int l=gi(),r=gi(),k=gi();
                int x=query(1,n,root[l-1],root[r],k);
                printf("%d
    ",p[x]);
            }
            return 0;
        }
    
    
  • 相关阅读:
    塔 · 第 二 条 约 定
    nyoj 325
    塔 · 第 一 条 约 定
    大一上
    Django之ORM
    mysql概念
    数据库索引
    使用pymysql进行数据库的增删改查
    sql注入攻击
    pymysql
  • 原文地址:https://www.cnblogs.com/ljq-despair/p/8639345.html
Copyright © 2020-2023  润新知