problem
给定一个长度为 n n n 的序列 { a n } \{a_n\} {an},有 q q q 个询问,每次询问区间 [ l , r ] [l,r] [l,r] 内 a i × a i a_i\times a_i ai×ai 出现次数的最大值。
数据范围: 1 ≤ n , q ≤ 1 0 5 1\le n,q\le 10^5 1≤n,q≤105, 1 ≤ a i ≤ 1 0 9 1\le a_i\le 10^9 1≤ai≤109。
solution
这道题是回滚莫队的板子题,可以参考这篇博客。
发现这道题加点很好做,但是删点就不容易了,因为我们不太容易快速找到次小值。
所以我们有一种只加不减的莫队:
- 对原序列进行分块,以左端点所在的块升序为第一关键字,以右端点升序为第二关键字来排序。
- 对于处理所有左端点在块 T T T 内的询问,我们先将莫队区间左端点初始化为 R [ T ] + 1 R[T]+1 R[T]+1,右端点初始化为 R [ T ] R[T] R[T],这是一个空区间。
- 对于左右端点在同一个块中的询问,我们直接暴力扫描回答即可。
- 对于左右端点不在同一个块中的所有询问,由于其右端点升序,我们对右端点只做加点操作,总共最多加点 n n n 次。
- 对于左右端点不在同一个块中的所有询问,其左端点是可能乱序的,我们每一次从 R [ T ] + 1 R[T]+1 R[T]+1 的位置出发,只做加点操作,到达询问位置即可,每一个询问最多加 n \sqrt n n 次。回答完询问后,我们撤销本次移动左端点的所有改动,使左端点回到 R [ T ] + 1 R[T]+1 R[T]+1 的位置
- 按照相同的方式处理下一块。
时间复杂度 O ( n n ) O(n\sqrt n) O(nn)。
code
#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
#define ll long long
using namespace std;
int n,q,S,num;
int a[N],d[N],in[N],cnt[N];
struct query{
int l,r,id;
}Q[N];
ll now,ans[N];
bool operator<(const query &p,const query &q){
return (in[p.l]==in[q.l])?p.r<q.r:in[p.l]<in[q.l];
}
void discrete(){
sort(d+1,d+n+1);
int tot=unique(d+1,d+n+1)-(d+1);
for(int i=1;i<=n;++i) a[i]=lower_bound(d+1,d+tot+1,a[i])-d;
}
int Times[N];
ll Get(int l,int r){
ll ans=0;
memset(Times,0,sizeof(Times));
for(int i=l;i<=r;++i){
Times[a[i]]++;
ans=max(ans,(ll)d[a[i]]*Times[a[i]]);
}
return ans;
}
void Add(int pos){
cnt[a[pos]]++;
now=max(now,(ll)d[a[pos]]*cnt[a[pos]]);
}
void Del(int pos){
cnt[a[pos]]--;
}
int solve(int i,int id){
memset(cnt,0,sizeof(cnt));
int lim=min(n,S*id),L=lim+1,R=lim;now=0;
for(;in[Q[i].l]==id;++i){
if(in[Q[i].l]==in[Q[i].r]){
ans[Q[i].id]=Get(Q[i].l,Q[i].r);continue;
}
while(R<Q[i].r) Add(++R);
ll tmp=now;
while(L>Q[i].l) Add(--L);
ans[Q[i].id]=now;
while(L<=lim) Del(L++);
now=tmp;
}
return i;
}
int main(){
scanf("%d%d",&n,&q),S=sqrt(n);
for(int i=1;i<=n;++i){
scanf("%d",&a[i]),d[i]=a[i];
in[i]=(i-1)/S+1,num=max(num,in[i]);
}
discrete();
for(int i=1;i<=q;++i){
scanf("%d%d",&Q[i].l,&Q[i].r),Q[i].id=i;
}
sort(Q+1,Q+q+1);
for(int i=1,id=1;id<=num;++id) i=solve(i,id);
for(int i=1;i<=q;++i) printf("%lld\n",ans[i]);
return 0;
}