【题解】Luogu P5338 [TJOI2019]甲苯先生的滚榜

原题传送门

这题明显可以平衡树直接大力整，所以我要说一下线段树+树状数组的做法

实际线段树+树状数组的做法也很暴力

我们先用树状数组维护每个ac数量有多少个队伍。这样就能快速求出有多少队伍ac数比现在这个队伍ac数多

我们再用\(n\)棵动态开点的线段树，第\(i\)棵线段树维护的是ac数为\(i\)的队伍的罚时情况。当一个队伍ac数为\(x\)罚时为\(t\)时，就在第\(x\)棵线段树\(t\)上加一。这样就能快速求出有多少队伍ac数与现在这个队伍ac数相同且罚时更少

当一个队伍过了一题后就在线段树和树状数组中正常修改即可

#include <bits/stdc++.h>
#define M 1000005
#define N 150005
#define ML 1500005
#define uint unsigned int
#define getchar nc
using namespace std;
inline char nc(){
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    register int x=0,f=1;register char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
    return x*f;
}
inline void write(register int x)
{
    if(!x)putchar('0');if(x<0)x=-x,putchar('-');
    static int sta[20];register int tot=0;
    while(x)sta[tot++]=x%10,x/=10;
    while(tot)putchar(sta[--tot]+48);
}
uint randNum(uint& seed,uint last,const uint mod)
{
    seed=17*seed+last;
    return seed%mod+1;
}
struct bit{
    int tr[N];
    inline void init()
    {
        memset(tr,0,sizeof(tr));
    }
    inline void update(register int pos,register int val)
    {
        for(register int i=pos;i<N;i+=i&(-i))
            tr[i]+=val;
    }
    inline int query(register int pos)
    {
        int res=0;
        for(register int i=pos;i;i-=i&(-i))
            res+=tr[i];
        return res;
    }
}tr1;
struct segt{
    struct node{
        int ls,rs,sum;
    }tr[M*40];
    int tot,root[N];
    inline void init()
    {
        memset(root,0,sizeof(root));
        tot=0;
    }
    inline void update(register int &rt,register int l,register int r,register int pos,register int val)
    {
        if(!rt)
        { 
            rt=++tot;
            tr[rt].ls=tr[rt].rs=tr[rt].sum=0;
        } 
        if(l==r)
        {
            tr[rt].sum+=val;
            return;
        }
        int mid=l+r>>1;
        if(pos<=mid)
            update(tr[rt].ls,l,mid,pos,val);
        else    
            update(tr[rt].rs,mid+1,r,pos,val);
        tr[rt].sum=tr[tr[rt].ls].sum+tr[tr[rt].rs].sum;
    }
    inline int query(register int &rt,register int l,register int r,register int L,register int R)
    {
        if(!rt)
            return 0;
        if(L<=l&&r<=R)
            return tr[rt].sum;
        int mid=l+r>>1,res=0;
        if(L<=mid)
            res+=query(tr[rt].ls,l,mid,L,R);
        if(R>mid)
            res+=query(tr[rt].rs,mid+1,r,L,R);
        return res;
    }
}tr2;
uint seed,last;
int T,n,m,tim[N],num[N];
int main()
{
    T=read();
    last=7;
    while(T--)
    {
        m=read(),n=read(),seed=read();
        memset(num,0,sizeof(num));
        memset(tim,0,sizeof(tim));
        tr1.init(),tr2.init();
        for(register int i=1;i<=m;++i)
            num[i]=1,tim[i]=1;
        tr1.update(1,m),tr2.update(tr2.root[1],1,ML,1,m);
        for(register int i=1;i<=n;++i)
        {
            int p=randNum(seed,last,m),v=randNum(seed,last,m);
            tr1.update(num[p],-1);
            tr2.update(tr2.root[num[p]],1,ML,tim[p],-1);
            ++num[p],tim[p]+=v;
            tr1.update(num[p],1);
            tr2.update(tr2.root[num[p]],1,ML,tim[p],1);
            int res=m-tr1.query(num[p]);
            res+=tr2.query(tr2.root[num[p]],1,ML,1,tim[p]-1);
            last=res;
            write(res),puts("");
        }
    }
    return 0;
}

转载于:https://www.cnblogs.com/yzhang-rp-inf/p/10954797.html