【bzoj4241】历史研究 分块

PoPoQQQ大神的题解：

http://blog.csdn.net/popoqqq/article/details/47809173

一开始自己想了一种莫队的方法，结果发现复杂度错了，结果在网上找到了在线的做法。

分块后，处理出mx[i][j]表示第i块到第j块的答案，cnt[i][j]表示前i块数字j的个数，注意先离散化。

然后每次查询，把块外的部分单独处理，块内的部分直接写答案就可以了。

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iostream>
#define maxn 100100 

using namespace std;

struct yts1
{
	int x,id;
}a[maxn];

int b[maxn],bel[maxn];
int n,m,block,tot;
int L[400],R[400];
long long mx[400][400];
int cnt[400][maxn];
long long ans;
int num[maxn];
int st[maxn],top;

bool cmp(yts1 x,yts1 y)
{
	return x.x<y.x;
}

long long calc(int l,int r)
{
	long long ans=0;
	top=0;
	for (int i=l;i<=r;i++)
	{
		if (!num[b[i]]) st[++top]=b[i];
		num[b[i]]++;
		ans=max(ans,(long long)num[b[i]]*a[b[i]].x);
	}
	while (top) num[st[top--]]=0;
	return ans;
}

int main()
{
	//freopen("data.in","r",stdin);
	//freopen("data.out","w",stdout);
	scanf("%d%d",&n,&m);
	for (int i=1;i<=n;i++)
	{
		scanf("%d",&a[i].x);
		a[i].id=i;
	}
	sort(a+1,a+n+1,cmp);
	b[a[1].id]=1;
	for (int i=2;i<=n;i++) if (a[i].x==a[i-1].x) b[a[i].id]=b[a[i-1].id]; else b[a[i].id]=i;
	block=sqrt(n);tot=(n-1)/block+1;
	for (int i=1;i<=n;i++)
	{
		bel[i]=(i-1)/block+1;
		if (!L[bel[i]]) L[bel[i]]=i;
		R[bel[i]]=i;
	}
	for (int i=1;i<=n;i++) cnt[bel[i]][b[i]]++;
	for (int i=2;i<=tot;i++)
	  for (int j=1;j<=n;j++)
	    cnt[i][j]+=cnt[i-1][j];
	for (int i=1;i<=tot;i++)
	{
		long long ans=0;
		memset(num,0,sizeof(num));
		for (int j=L[i];j<=n;j++)
		{
			num[b[j]]++;
			ans=max(ans,(long long)num[b[j]]*a[b[j]].x);
			if (j==R[bel[j]]) mx[i][bel[j]]=ans;
		}
	}
	memset(num,0,sizeof(num));
	for (int i=1;i<=m;i++)
	{
		int l,r;
		scanf("%d%d",&l,&r);
		ans=0;
		if (bel[l]==bel[r]) printf("%lld\n",calc(l,r));
		else
		{
			ans=mx[bel[l]+1][bel[r]-1];
			top=0;
			for (int i=l;i<=R[bel[l]];i++)
			  if (!num[b[i]]) st[++top]=b[i],num[b[i]]=cnt[bel[r]-1][b[i]]-cnt[bel[l]][b[i]];
			for (int i=L[bel[r]];i<=r;i++)
			  if (!num[b[i]]) st[++top]=b[i],num[b[i]]=cnt[bel[r]-1][b[i]]-cnt[bel[l]][b[i]];
			for (int i=l;i<=R[bel[l]];i++) num[b[i]]++;
			for (int i=L[bel[r]];i<=r;i++) num[b[i]]++;
			while (top)
			{
				ans=max(ans,(long long)a[st[top]].x*num[st[top]]);
				num[st[top--]]=0;
			}
			printf("%lld\n",ans);
		}
	}
	return 0;
}