回滚莫队
题目描述
IOI国历史研究的第一人——JOI教授,最近获得了一份被认为是古代IOI国的住民写下的日记。JOI教授为了通过这份日记来研究古代IOI国的生活,开始着手调查日记中记载的事件。
日记中记录了连续N天发生的时间,大约每天发生一件。
事件有种类之分。第i天(1<=i<=N)发生的事件的种类用一个整数Xi表示,Xi越大,事件的规模就越大。
JOI教授决定用如下的方法分析这些日记:
1.选择日记中连续的一些天作为分析的时间段
2.事件种类t的重要度为t×(这段时间内重要度为t的事件数)
3.计算出所有事件种类的重要度,输出其中的最大值
现在你被要求制作一个帮助教授分析的程序,每次给出分析的区间,你需要输出重要度的最大值。
输入格式
第一行两个空格分隔的整数N和Q,表示日记一共记录了N天,询问有Q次。
接下来一行N个空格分隔的整数X1…XN,Xi表示第i天发生的事件的种类
接下来Q行,第i行(1<=i<=Q)有两个空格分隔整数Ai和Bi,表示第i次询问的区间为[Ai,Bi]。
普通莫队适用于区间的增大和缩小都好处理的情况,但有的情况删除并不好处理,回滚莫队很好地解决了这个问题。
其关键在于,分块处理,左右端点在一个块中则暴力处理,否则由于莫队的排序在左节点在同一块时,有节点是递增的,我们只要递增处理右节点,则右侧没有删除;然后左节点没处理一个问题就回溯到当前块尾,则左侧也没有删除操作,左右都是
O
(
n
n
)
O(n\sqrt n)
O(nn)
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long ll;
const ll maxn=1e5+10;
ll ans[maxn];
ll a[maxn];
ll b[maxn];
ll pos[maxn];
ll be[maxn];
ll low[maxn];
ll high[maxn];
ll cnt[maxn];//用于莫队
ll cnt2[maxn];//用于左右段点在同一块时的暴力
struct Q{
ll l,r,id;
friend bool operator <(Q a,Q b){
return be[a.l]^be[b.l]?be[a.l]<be[b.l]:a.r<b.r;
}
}q[maxn];
int main(){
ll n,m;
scanf("%lld%lld",&n,&m);
ll sz=sqrt(n);
ll bnum=ceil((double)n/sz);//块的数量
for(ll i=1;i<=bnum;i++){
low[i]=(i-1)*sz+1;
high[i]=i*sz;
for(ll j=low[i];j<=high[i];j++){
be[j]=i;
}
}
high[bnum]=n;
for(ll i=1;i<=n;i++){
scanf("%lld",&a[i]);
b[i]=a[i];
}
sort(b+1,b+1+n);
ll tot=unique(b+1,b+1+n)-b-1;
for(ll i=1;i<=n;i++){
pos[i]=lower_bound(b+1,b+1+n,a[i])-b;
}
for(ll i=1;i<=m;i++){
scanf("%lld%lld",&q[i].l,&q[i].r);
q[i].id=i;
}
sort(q+1,q+1+m);
ll j=1;
for(ll i=1;i<=bnum;i++){
ll l=high[i]+1;
ll r=high[i];
ll now=0;
memset(cnt,0,sizeof(cnt));
for(;be[q[j].l]==i;j++){
ll ql=q[j].l,qr=q[j].r;
ll tmp;
if(be[ql]==be[qr]){
tmp=0;
for(ll k=ql;k<=qr;k++){
cnt2[pos[k]]=0;
}
for(ll k=ql;k<=qr;k++){
++cnt2[pos[k]];
tmp=max(tmp,cnt2[pos[k]]*a[k]);
}
ans[q[j].id]=tmp;
}
else{
while(r<qr){
++r;
++cnt[pos[r]];
now=max(now,cnt[pos[r]]*a[r]);
}
tmp=now;
while(l>ql){
--l;
++cnt[pos[l]];
now=max(now,cnt[pos[l]]*a[l]);
}
ans[q[j].id]=now;
while(l<high[i]+1){
cnt[pos[l]]--;
l++;
}
now=tmp;
}
}
}
for(ll i=1;i<=m;i++){
printf("%lld\n",ans[i]);
}
}