atcoder2014 joi训练营 C 歴史の研究 回滚莫队

https://atcoder.jp/contests/joisc2014/tasks/joisc2014_c

经典找最大值,删除的时候找次大找不了,所以我们需要用不删除的回滚莫队,每次操作都从左端点所在块的最右边开始向左移动到左端点然后再回退,由于同一左端点块内右端点都是严格向右的,所以一个块只会向右n,只有sqrt(n)个块,每个询问只要扫左端点块一遍,长度为sqrt(n),所以左端点移动总数是qsqrt(n)

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int maxl=1e5+10;

int n,m,len,tot,cnt;
int a[maxl],b[maxl],c[maxl],num[maxl],tnum[maxl];
int bel[maxl],clr[maxl];
ll ans[maxl];
struct qry
{
	int l,r,id;
}q[maxl];

inline bool cmp(const qry &a,const qry &b)
{
	return bel[a.l]!=bel[b.l]?bel[a.l]<bel[b.l]:a.r<b.r;
}

inline void prework()
{
	scanf("%d%d",&n,&m);
	len=sqrt(n);
	for(int i=1;i<=n;i++)
		scanf("%d",&a[i]),b[i]=a[i],bel[i]=(i-1)/len+1;
	sort(b+1,b+1+n);
	tot=unique(b+1,b+1+n)-b-1;
	for(int i=1;i<=n;i++)
		c[i]=lower_bound(b+1,b+1+tot,a[i])-b;
	for(int i=1;i<=m;i++)
		scanf("%d%d",&q[i].l,&q[i].r),q[i].id=i;
	sort(q+1,q+1+m,cmp);
}

inline ll bf(int l,int r)
{
	ll ret=0;
	for(int i=l;i<=r;i++)
	{
		++tnum[c[i]];	
		ret=max(ret,1ll*tnum[c[i]]*a[i]);
	}
	for(int i=l;i<=r;i++)
		tnum[c[i]]=0;
	return ret;
}

inline void mainwork()
{
	int id=1,l,r,br;ll mx,tmp;
	for(int i=1;i<=bel[n];i++)
	{
		br=min(i*len,n);l=br+1;r=br;mx=0;cnt=0;
		while(bel[q[id].l]==i && id<=m)
		{
			if(bel[q[id].r]==i)
				ans[q[id].id]=bf(q[id].l,q[id].r);
			else
			{
				while(r<q[id].r)
				{
					++r;
					if(!num[c[r]])
						clr[++cnt]=c[r];
					++num[c[r]];
					mx=max(1ll*a[r]*num[c[r]],mx);
				}
				tmp=mx;
				while(l>q[id].l)
				{
					--l;
					++num[c[l]];
					mx=max(1ll*a[l]*num[c[l]],mx);
				}
				ans[q[id].id]=mx;
				while(l<br+1)
				{
					--num[c[l]];
					l++;
				}
				mx=tmp;
				
			}
			id++;
		}
		for(int j=1;j<=cnt;j++)
			num[clr[j]]=0;
	}
}

inline void print()
{
	for(int i=1;i<=m;i++)
		printf("%lld\n",ans[i]);
}

int main()
{
	prework();
	mainwork();
	print();
	return 0;
}

猜你喜欢

转载自blog.csdn.net/liufengwei1/article/details/109152423
今日推荐