[BZOJ4241] 回滚莫队

[BZOJ4241]

• 给出n个数，每次询问区间[L,R]内每个数*该数出现次数的最大值
• 使用回滚莫队
• 回滚莫队用来维护不可撤销信息比如最大值
• 在同一个块内右端点递增左端点每次移动到该块的最右边再插入元素
• 当查询的左端点在一个块内时右端点移动O(n)，总共有sqrt(n)个块，总时间复杂度为O(n*sqrt(n))
• 左端点每查询一次最多是O(sqrt(n))的时间复杂度，总时间复杂度为O(m*sqrt(n))
• 故回滚莫队总时间复杂度为O(n*sqrt(n))
• 注意写代码时的一些技巧

Code

---

#include<bits/stdc++.h>
#define rep(i,a,b) for(int i=(a);i<=(b);i++)
#define per(i,a,b) for(int i=(a);i>=(b);i--)
#define ll long long
using namespace std;
const int N=1e5+100;
struct Q{int l,r,id;}q[N];
int bl[N],v[N],a[N],b[N],blo,st,l,r,n,m;
ll sum[N],num[N],AC[N],nowans,lastans;
bool cmp(Q a,Q b){return (bl[a.l]==bl[b.l]&&a.r<b.r)||(bl[a.l]<bl[b.l]);}
void add(int x,ll &ans){
       
    sum[a[x]]+=b[a[x]];
    ans=max(ans,sum[a[x]]);
}
void del(int x){
    sum[a[x]]-=b[a[x]];
}
int main()
{
    //freopen("a.in","r",stdin);
    scanf("%d%d",&n,&m); blo=sqrt(n);
    rep(i,1,n)scanf("%d",&a[i]),b[i]=a[i];
    sort(b+1,b+n+1);int cnt=unique(b+1,b+n+1)-b-1;
    rep(i,1,n)a[i]=lower_bound(b+1,b+cnt+1,a[i])-b;
    rep(i,1,n)bl[i]=(i-1)/blo+1;
    rep(i,1,m)scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
    sort(q+1,q+m+1,cmp);
    rep(i,1,m){
        if(bl[q[i].l]!=bl[q[i-1].l]){
            memset(sum,0,sizeof(sum));
            st=bl[q[i].l]*blo;
            nowans=lastans=0,l=st,r=l-1;
        }
        if(bl[q[i].l]==bl[q[i].r]){
            rep(j,q[i].l,q[i].r)num[a[j]]+=b[a[j]],AC[q[i].id]=max(AC[q[i].id],num[a[j]]);
            rep(j,q[i].l,q[i].r)num[a[j]]=0;
            continue;
        }
        while(r<q[i].r)add(++r,lastans);
        nowans=lastans;
        while(l>q[i].l)add(--l,nowans);
        AC[q[i].id]=nowans;
        while(l<st)del(l++);
    }
    rep(i,1,m)printf("%lld\n",AC[i]);
    return 0;
}