主席树 (动态)图文讲解让你一次就懂 zoj2112为例

版权声明:编写不易,转载请注明出处,谢谢。 https://blog.csdn.net/WilliamSun0122/article/details/77885781

主席树(动态)

学动态主席树之前一定要先会静态主席树。
静态主席树:http://blog.csdn.net/williamsun0122/article/details/77871278

参考博客:http://www.cnblogs.com/Empress/p/4659824.html
这个博客讲的是对的,不过它的图和代码好像是错的,虽然能过,但应该是数据不够强,具体怎么错的它的博客下面评论有。

其实静态主席树我们弄清楚之后,动态的可以很快学会。因为动态主席树就是在静态主席树的基础上增加了一批用树状数组思维维护的线段树。
我们以下面的例子讲解。
5 3
3 2 1 4 7
Q 1 4 3 询问区间[1,4]第3小数
C 2 6 把第2个数变为6
Q 2 5 3 询问区间[2,5]第3小数

n是原序列个数
T[i]表示第i棵线段树的根节点编号
S[i]表示树状数组思维建的第i棵线段树的根节点编号
L[i]表示节点i的左子节点编号
R[i]表示节点i的右子节点编号
sum[i]表示节点i对应区间中数的个数。

这里离散化建树过程和静态主席树有一点不同,我们必须把所有询问先存起来并且把改变的数也加入到原序列中再离散化建树,会导致空间复杂度和静态有所区别(之前讲静态的时候提过)。所以这里我们离散化后序列为3 2 1 4 6 5分别对应原序列的3 2 1 4 7和改变后的6。

之后同静态一样建空树,按原序列前缀建树,相信不用我说了。(画图有点麻烦,我借鉴参考博客的图,这个图是没错的)

接下来就是重点了,对于题目给出的修改操作,我们新建一批线段树来记录更新,这些线段树以树状数组的思维来维护。
一开始,S[0]、S[1]、S[2]、S[3]、S[4]、S[5] (注意一共有n+1个 即 0到n)(树状数组的每个节点)这些都与T[0]相同(也就是每个节点建了一棵空树)。
对于C 2 6 这个操作, 我们只需要减去一个2,加上一个5(对应改变后的6)即可。
这个更新我们按树状数组的思想更新,比如这里的减2,我们要从i=2(原序列中第2个数2在离散化后序列中的位置)即S[2]开始更新,并往上lowbit(i)直到大于5,这里我们会更新S[2]和S[4]。
边看图边理解(这个图最后应该是在节点5那里减1)

对于加5同样是从S[2]开始更新
(这个图最后应该是在节点10那里加1)

这样我们查询的时候T[]和静态一样,再按树状数组的思维加上S[]就可算出每个节点对应区间中数的个数,再按静态的思想查询即可。

对于原序列n个数,m次询问
空间复杂度(我的理解)
因为这里我们要加入询问中数离散化后再建树,所以建树4*(n+m)
按原序列更新T[],n log2(n+m)
每次询问更新S[],2 log2n (按树状数组最多更新的节点) log2(n+m) (每次最多更新线段树中的节点)
加起来即可,不过你真这么开数组的话肯定会MLE。

以zoj2112为例,参考博客开的范围是2*(n+m) log2(n+m) (大概,我是强行解释,我范围开大会Segmentation fault,开小会RE)
我猜大概是因为如果每次询问都是修改操作的话按静态中的开应该会是(n+m) log2(n+m) 还要加上一个S[],所以要*2。

时间复杂度(我的理解)
m log2n log2(n+m)

ZOJ2112

题意
给你n(<=5e4)个数,m(<=1e4)次询问。
Q i j k 为问区间[i,j]第k小的数
C i t 为把原序列中第i个数变为t

题解
裸题,边看代码边理解上面思想

#include <bits/stdc++.h>
using namespace std;

const int maxn = 6e4+5;
const int maxm = 1e4+5;
int T[maxn],S[maxn],L[maxn*32],R[maxn*32],sum[maxn*32];
int sz[maxn],h[maxn];
int ul[maxn],ur[maxn];
int tot,num,n,q;

struct node{
    int l,r,k;
    bool flag; //ture代表Q,false代表C
}Q[maxm];  //存储询问

void build(int& rt,int l,int r)
{
    rt = ++tot;
    sum[rt]=0;
    if(l==r) return;
    int mid = (l+r)>>1;
    build(L[rt],l,mid);
    build(R[rt],mid+1,r);
}

void update(int& rt,int pre,int l,int r,int x,int val)
{
    rt = ++tot;
    L[rt] = L[pre];
    R[rt] = R[pre];
    sum[rt] = sum[pre]+val;
    if(l==r) return;
    int mid = (l+r)>>1;
    if(x<=mid) update(L[rt],L[pre],l,mid,x,val);
    else update(R[rt],R[pre],mid+1,r,x,val);
}

int lowbit(int x)
{
    return x&(-x);
}

void add(int x,int val)
{
    int res = lower_bound(h+1,h+1+num,sz[x])-h;
    while(x<=n)
    {
        update(S[x],S[x],1,num,res,val);
        x += lowbit(x);
    }
}

int Sum(int x,bool flag)
{
    int res=0;
    while(x>0)
    {
        if(flag) res += sum[L[ur[x]]];
        else res += sum[L[ul[x]]];
        x -= lowbit(x);
    }
    return res;
}

int query(int s,int e,int ts,int te,int l,int r,int k)
{
    if(l==r) return l;
    int mid = (l+r)>>1;
    int res = Sum(e,true)-Sum(s,false)+sum[L[te]]-sum[L[ts]];
    if(k<=res)
    {
        for(int i=e;i;i-=lowbit(i)) ur[i] = L[ur[i]];
        for(int i=s;i;i-=lowbit(i)) ul[i] = L[ul[i]];
        return query(s,e,L[ts],L[te],l,mid,k);
    }
    else
    {
        for(int i=e;i;i-=lowbit(i)) ur[i] = R[ur[i]];
        for(int i=s;i;i-=lowbit(i)) ul[i] = R[ul[i]];
        return query(s,e,R[ts],R[te],mid+1,r,k-res);
    }
}


int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        char str[5];
        num=0;
        scanf("%d%d",&n,&q);
        for(int i=1;i<=n;i++) scanf("%d",sz+i),h[++num]=sz[i];
        for(int i=1;i<=q;i++)
        {
            scanf("%s",str);
            if(str[0]=='Q')
            {
                scanf("%d%d%d",&Q[i].l,&Q[i].r,&Q[i].k);
                Q[i].flag=true;
            }
            else
            {
                scanf("%d%d",&Q[i].l,&Q[i].r);
                Q[i].flag=false;
                h[++num]=Q[i].r;
            }
        }
        sort(h+1,h+1+num);
        int tmp = unique(h+1,h+1+num)-h-1;
        num = tmp;
        tot=0;
        build(T[0],1,num);
        for(int i=1;i<=n;i++) update(T[i],T[i-1],1,num,lower_bound(h+1,h+1+num,sz[i])-h,1);
        for(int i=1;i<=n;i++) S[i] = T[0];
        for(int i=1;i<=q;i++)
        {
            if(Q[i].flag)
            {
                for(int j=Q[i].r;j;j-=lowbit(j)) ur[j] = S[j];
                for(int j=Q[i].l-1;j;j-=lowbit(j)) ul[j] = S[j];
                printf("%d\n",h[query(Q[i].l-1,Q[i].r,T[Q[i].l-1],T[Q[i].r],1,num,Q[i].k)]);
            }
            else
            {
                add(Q[i].l,-1);
                sz[Q[i].l] = Q[i].r;
                add(Q[i].l,1);
            }
        }
    }
    return 0;
}

比较抽象,真不太好通过文字表达清楚,结合代码多看几遍讲解应该还是能搞清楚的,切记不要浮躁。

猜你喜欢

转载自blog.csdn.net/WilliamSun0122/article/details/77885781