分块大法好,这题说白了区间加权众数
代码里有部分重点注释
//By AcerMo
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define lli long long int
using namespace std;
const int M=1e5+50;
int n,m,block;
int cnt[333][M],num[M];
int a[M],bloc[M],b[M],sta[M];
lli f[333][M];
inline int read()
{
int x=0;char ch=getchar();
while (ch>'9'||ch<'0') ch=getchar();
while (ch<='9'&&ch>='0') x=x*10+ch-'0',ch=getchar();
return x;
}
inline lli get(int l,int r)
{
lli ans=f[bloc[l]+1][r];//先处理整块
int top=0,tmp=lower_bound(bloc+1,bloc+n+1,bloc[r])-bloc;//找到r所在块的开头
for (int i=tmp;i<=r;i++) num[a[i]]++,sta[++top]=a[i];//记录哪些出现过
tmp=lower_bound(bloc+1,bloc+n+1,bloc[l]+1)-bloc;//l所在块的末尾
for (int i=l;i<tmp;i++)
{
sta[++top]=a[i];num[a[i]]++;
ans=max(ans,(lli)(cnt[bloc[l]+1][a[i]]+num[a[i]]-cnt[bloc[r]][a[i]])*b[a[i]]);//边角余料
}
//边角余料就是两端不是整块的部分,需要单独计算
for (int i=1;i<=top;i++) num[sta[i]]=0;//清空
return ans;
}
signed main()
{
n=read();m=read();block=sqrt(n);int tot=0;
for (int i=1;i<=n;i++) a[i]=read(),b[i]=a[i];
for (int i=1;i<=n;i++) bloc[i]=(i-1)/block+1;
sort(b+1,b+n+1);int q=unique(b+1,b+n+1)-b-1;//好像是去重来着
for (int i=1;i<=n;i++) a[i]=lower_bound(b+1,b+q+1,a[i])-b;//找到每个元素的排名,离散化?
if (n%block) tot=n/block+1;else tot=n/block;
for (int i=1;i<=tot;i++)
{
lli ans=0;
for (int k=lower_bound(bloc+1,bloc+n+1,i)-bloc;k<=n;k++)//k从i所在块的 开头一直处理到所有数末尾
cnt[i][a[k]]++,ans=max(ans,(lli)cnt[i][a[k]]*b[a[k]]),f[i][k]=ans;//预处理块内答案
}
while (m--)
{
int l=read(),r=read();
lli ans=get(l,r);//long long
cout<<ans<<endl;
}
return 0;
}