历史研究（回滚莫队）（add）

Posted 2021-12-06 thusloop

历史研究
题意：给你一个长度为n的数组 和 q个询问 ，每次询问一个区间 l,r 内重要度最大的数字，要求 输出其重要度。一个数字 x重要度的定义为 x乘上x 在区间内出现的次数。

//#pragma GCC optimize(2)
//#pragma GCC optimize(3,"Ofast","inline")
#include<bits/stdc++.h>
#define int long long
#define fi first
#define se second
#define pb push_back
#define pii pair<int,int>
#define ios ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
using namespace std;
const int inf=8e18;
const int maxn=2e5+100;
int a[maxn],b[maxn],cnt[maxn],L[maxn],R[maxn],pos[maxn],cnt_tp[maxn],ans[maxn];
map<int,int>mp;
int res;
struct node
{
	int l,r,id;
} t[maxn];
bool cmp(node a,node b)
{
	if(pos[a.l]==pos[b.l])return a.r<b.r;
	return pos[a.l]<pos[b.l];
}
void add(int x,int &tp)
{
	cnt[x]++;
	tp=max(tp,cnt[x]*b[x]);
}
void del(int x)
{
	cnt[x]--;
}
signed main()
{
	int n,q;
	cin>>n>>q;
	int sz=sqrt(n);
	for(int i=1; i<=n; i++)
	{
		cin>>a[i];
		mp[a[i]]=1;
		pos[i]=(i-1)/sz+1;
		if(!L[pos[i]])L[pos[i]]=i;
		R[pos[i]]=max(R[pos[i]],i);
	}
	int mx=0;
	for(auto it:mp) mp[it.fi]=++mx;
	for(int i=1; i<=n; i++) b[mp[a[i]]]=a[i];
	for(int i=1; i<=n; i++) a[i]=mp[a[i]]; 
	for(int i=1; i<=q; i++)
	{
		int l,r;
		cin>>l>>r;
		t[i]= {l,r,i};
	}
	sort(t+1,t+q+1,cmp);
	int l=1,r=0,last=0,ll;
	for(int i=1; i<=q; i++)
	{
		if(pos[t[i].l]==pos[t[i].r])
		{
			int tp=0;
			for(int j=t[i].l; j<=t[i].r; j++)
			{
				cnt_tp[a[j]]++;
				tp=max(tp,cnt_tp[a[j]]*b[a[j]]);
			}
			ans[t[i].id]=tp;
			for(int j=t[i].l; j<=t[i].r; j++)cnt_tp[a[j]]--;
			continue;
		}
		if(pos[t[i].l]!=last)
		{
			while(l<R[pos[t[i].l]]+1) del(a[l++]);
			while(r>R[pos[t[i].l]]) del(a[r--]);
			res=0;
			last=pos[t[i].l];
		}
		while(r<t[i].r) add(a[++r],res);
		ll=l;
		int tp=res;
		while(ll>t[i].l) add(a[--ll],tp);
		ans[t[i].id]=tp;
		while(ll<l) del(a[ll++]);
	}
	for(int i=1; i<=q; i++)
	{
		cout<<ans[i]<<"\\n";
	}
}