主席树总结-CSDN博客

PS：待修改主席树不会，树套树从来写不动

其实很早就学过这个东西了，基本的思路也可以理解，最基本的应用就是用来求区间第K大

贴一下我的主席树模板

#include<iostream>
#include<cstdio>
#include<algorithm>
#define M 200010
#define ls ch[node][0]
#define rs ch[node][1]
using namespace std;

int n,m,cnt,sz;
int a[M],b[M],rt[M],sum[M<<5],ch[M<<5][2];

int get(int x)
{
    int l=1,r=cnt;
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(b[mid]==x) return mid;
        if(b[mid]>x) r=mid-1;
        else l=mid+1;
    }
}
void build(int &node,int l,int r)
{
    node=++sz;
    if(l==r) return;
    int mid=(l+r)/2;
    build(ls,l,mid),build(rs,mid+1,r);
}

void insert(int &node,int pre,int l,int r,int x)
{
    node=++sz;
    sum[node]=sum[pre]+1,ls=ch[pre][0],rs=ch[pre][1];
    if(l==r) return;
    int mid=(l+r)/2;
    if(x<=mid) insert(ls,ch[pre][0],l,mid,x);
    else insert(rs,ch[pre][1],mid+1,r,x);
}

int query(int l1,int r1,int l,int r,int k)
{
    if(l==r) return l;
    int now=sum[ch[l1][0]]-sum[ch[r1][0]],mid=(l+r)/2;
    if(k<=now) return query(ch[l1][0],ch[r1][0],l,mid,k);
    else return query(ch[l1][1],ch[r1][1],mid+1,r,k-now);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        b[++cnt]=a[i];
    }
    sort(b+1,b+1+n);
    cnt=unique(b+1,b+1+n)-b-1;
    build(rt[0],1,n);
    for(int i=1;i<=n;i++) 
    {
        a[i]=get(a[i]);
        rt[i]=rt[i-1];
        insert(rt[i],rt[i-1],1,cnt,a[i]);
    }
    for(int i=1;i<=m;i++)
    {
        int l,r,k;scanf("%d%d%d",&l,&r,&k);
        printf("%d\n",b[query(rt[r],rt[l-1],1,cnt,k)]);
    }
    return 0;
}

View Code

一般来说理解了权值线段树再学主席树就很好理解了，因为权值线段树存在可减性，所以最基础的主席树就是在两棵权值线段树每个节点相减得到的新的线段树上进行操作，当然还会有别的变化，但无论怎么变这个线段树都应该是权值线段树

最近碰到了这样一个类似二维数点的题目然而我这么菜当然不会做啦：

有$n$条线段，每次查询在区间[$l$,$r$]内线段的条数

首先我们把线段都按照左端点排好序，这样问题就转化成:

求所有左端点在区间[$l$,$r$]内，其对应的右端点也在区间[$l$,$r$]内的点对的个数

我们可以对每个左端点建立主席树，维护其右端点的位置，查询就是求$Tree[r]$-$Tree[l-1]$构成的权值线段树在区间[$l$,$r$]内的$size$

是不是很简单~

下面放几道我最近做的主席树的题目

1.Count on a tree

题意：求树上两点之间的点权第k大值

对于一条链$u->v$上的信息，我们可以把它看成$(u->root)+(v->root)-LCA->root-fa_LCA->root$。这样我们把每个点在它的父节点的基础上建立主席树，查询的时候就不再是两棵树相减，而是四棵树相加减了

2.middle

题意：给定一个序列，每次查询左端点在[$a$,$b$]之间，右端点在[$c$,$d$]之间的中位数最大的子串

我们考虑转化成二分和0/1序列判定，即我们应该知道的是二分到每个数时序列的0/1状态，当我们加入最小的数时，序列的0/1状态为该位置是0，其他为1，再加入第二个数时，又在第二个数的位置上多了一个0。排序后每个数的二分序列都是在上一个数的二分序列的基础上增加一个0。那么思路就很明显了，用主席树维护每次在原来的基础上加修改一个位置即可。(因为这个主席树跟上边的不是很一样，感觉说的不是很清楚，自己画画图就很明显了)