【题解】Luogu CF86D Powerful array

原题传送门

裸的莫队啊,我博客里有对莫队较详细的介绍

这道题很简单,可以说是裸的模板

但是如何在已有的值上进行操作?

小学生应该都知道

那么转移就超级简单了qaq

inline void add(register int c)
{
	sum+=(num[c]+num[c]+1)*c;
	++num[c];
}
inline void del(register int c)
{
	--num[c];
	sum-=(num[c]+num[c]+1)*c;
}

要开long long(虽说我没试过int,但我觉得会炸int)

#include <bits/stdc++.h>
#define N 200005
using namespace std;
inline int read()
{
	register int x=0,f=1;register char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+ch-'0',ch=getchar();
	return x*f;
}
inline void write(register long long x)
{
	if(!x)putchar('0');if(x<0)x=-x,putchar('-');
	static int sta[25];int tot=0;
	while(x)sta[tot++]=x%10,x/=10;
	while(tot)putchar(sta[--tot]+48);
}
struct query{
	int l,r,id,bl;
}q[N];
int a[N],blocksize=0,num[1000005];
long long ans[N],sum=0;
inline void add(register int c)
{
	sum+=(num[c]+num[c]+1)*c;
	++num[c];
}
inline void del(register int c)
{
	--num[c];
	sum-=(num[c]+num[c]+1)*c;
}
inline bool cmp(register query a,register query b)
{
	return a.bl!=b.bl?a.l<b.l:(a.bl&1?a.r<b.r:a.r>b.r);
}
int main()
{
	int n=read(),m=read();
	blocksize=sqrt(n);
	for(register int i=1;i<=n;++i)
		a[i]=read();
	for(register int i=1;i<=m;++i)
	{
		int l=read(),r=read();
		q[i]=(query){l,r,i,(l-1)/blocksize+1};
	}
	sort(q+1,q+1+m,cmp);
	int l=1,r=0;
	for(register int i=1;i<=m;++i)
	{
		int ll=q[i].l,rr=q[i].r;
		while(ll<l)
			add(a[--l]);
		while(ll>l)
			del(a[l++]);
		while(rr>r)
			add(a[++r]);
		while(rr<r)
			del(a[r--]);
		ans[q[i].id]=sum;
	}
	for(register int i=1;i<=m;++i)
		write(ans[i]),printf("\n");
	return 0;
 } 
posted @ 2018-11-23 20:01  JSOI爆零珂学家yzhang  阅读(179)  评论(0编辑  收藏  举报