主席树（学习笔记，不含基础知识）

主席树的学习过程记录

这个只是做题的总结，基础知识啥的就直接贴链接了
主席树的基础知识
 
2021-8-25
P3919 【模板】可持久化线段树 1（可持久化数组）

这道题比较清楚地展示了主席树，也就是可持续化线段树的一个update（或者叫modify）的一个过程，在题目中叫做version（版本），挺好记的。

这道题完全可以拿来梳理主席树的基础概念，而且基本上写这道题的代码大同小异（拒绝cv战士）。

#include<iostream>
using namespace std;
#include<iomanip>
#include<string>
#include<algorithm>
#include<stack>
#include<map>
#include<queue>
#include<vector>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<cstdio>

typedef unsigned long long ULL;
typedef long long LL;
typedef long L;
typedef double DB;

const int maxn=1e6+10;
int n,m,a[maxn];

int cnt=0;
int rt[maxn<<5],ls[maxn<<5],rs[maxn<<5],sum[maxn<<5],val[maxn<<5];

inline void build(int &rt,int l,int r)
{
    rt=++cnt;
    if(l==r)
    {
        val[rt]=a[l];
        return ;
    }
    int mid=(l+r)>>1;
    build(ls[rt],l,mid);
    build(rs[rt],mid+1,r);
}

inline int update(int o,int l,int r,int x,int k)
{
    int oo=++cnt;ls[oo]=ls[o],rs[oo]=rs[o];
    if(l==r)
    {
        val[oo]=k;
        return oo;
    }
    int mid=(l+r)>>1;
    if( x<=mid )
        ls[oo]=update(ls[oo],l,mid,x,k);
    else
        rs[oo]=update(rs[oo],mid+1,r,x,k);
    return oo;
}

int query(int u,int l,int r,int x)
{
    if(l==r)
    {
        return val[u];
    }
    int mid=(l+r)>>1;
    if(x<=mid)
        return query(ls[u],l,mid,x);
    else
        return query(rs[u],mid+1,r,x);
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
        scanf("%d",&a[i]);
    build(rt[0],1,n);
    for(int i=1;i<=m;++i)
    {
        int v,op,pos,k;
        scanf("%d%d%d",&v,&op,&pos);
        if(op==1)
        {
            scanf("%d",&k);
            rt[i]=update(rt[v],1,n,pos,k);
        }
        else
        {
            printf("%d\n",query(rt[v],1,n,pos));
            rt[i]=rt[v];
        }
    }
    return 0;
}

 
 

P3834 【模板】可持久化线段树 2（主席树）

新手不建议直接写这道题，因为这道题对概念的运用就有点清晰了，同时还有离散化（好像数据结构里面很常见）的操作。

洛谷的题解里有大佬贴出了一个主席树的板子，可以当传家宝。

#include<iostream>
using namespace std;
#include<iomanip>
#include<string>
#include<algorithm>
#include<stack>
#include<map>
#include<queue>
#include<vector>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<cctype>
#include<cstdio>

typedef unsigned long long ULL;
typedef long long LL;
typedef long L;
typedef double DB;

const int maxn=2e5+10;
int n,m,lsh;
int a[maxn],b[maxn];

int cnt=0;
int rt[maxn<<5],ls[maxn<<5],rs[maxn<<5],sum[maxn<<5];

inline void build(int &rt,int l,int r)
{
    rt=++cnt;
    if(l==r)
        return ;
    int mid=(l+r)>>1;
    build(ls[rt],l,mid);
    build(rs[rt],mid+1,r);
}

int update(int o,int l,int r,int p)
{
    int oo=++cnt;
    ls[oo]=ls[o];rs[oo]=rs[o];sum[oo]=sum[o]+1;
    if(l==r)
        return oo;
    int mid=(l+r)>>1;
    if(p<=mid)
        ls[oo]=update(ls[oo],l,mid,p);
    else
        rs[oo]=update(rs[oo],mid+1,r,p);
    return oo;
}

int query(int u,int v,int l,int r,int k)
{
    int ret,mid=(l+r)>>1,x=sum[ls[v]]-sum[ls[u]];
    if(l==r)
        return l;
    if( x>=k )
        ret=query(ls[u],ls[v],l,mid,k);
    else
        ret=query(rs[u],rs[v],mid+1,r,k-x);
    return ret;
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;++i)
    {
        scanf("%d",&a[i]);
        b[i]=a[i];
    }
    sort(b+1,b+n+1);
    lsh=unique(b+1,b+n+1)-(b+1);

    build(rt[0],1,lsh);

    for(int i=1;i<=n;++i)
    {
        int p=lower_bound(b+1,b+lsh+1,a[i])-b;
        rt[i]=update(rt[i-1],1,lsh,p);
    }

    int l,r,k,ans;

    while(m--)
    {
        scanf("%d%d%d",&l,&r,&k);
        ans=query(rt[l-1],rt[r],1,lsh,k);
        printf("%d\n",b[ans]);
    }

    return 0;
}