BZOJ 3781 小B的询问莫队

最新推荐文章于 2019-02-17 13:09:02 发布

MyZhY

最新推荐文章于 2019-02-17 13:09:02 发布

阅读量264

点赞数

分类专栏：莫队

本文链接：https://blog.csdn.net/ThinFatty/article/details/74066335

版权

莫队专栏收录该内容

6 篇文章 1 订阅

订阅专栏

Description

小B有一个序列，包含N个1~K之间的整数。他一共有M个询问，每个询问给定一个区间[L..R]，求Sigma(c(i)^2)的值,其中i的值从1到K，其中c(i)表示数字i在[L..R]中的重复次数。小B请你帮助他回答询问。

Input

第一行，三个整数N、M、K。

第二行，N个整数，表示小B的序列。

接下来的M行，每行两个整数L、R。

Output

M行，每行一个整数，其中第i行的整数表示第i个询问的答案。

Sample Input

6 4 3
1 3 2 1 1 3
1 4
2 6
3 5
5 6

Sample Output

6
9
5
2

HINT

对于全部的数据，1<=N、M、K<=50000

Source

传送门

莫队的模板题目……

对于平方这个要求，

我们只要更新ans的时候，先减去原来的，再加上更新过的出现次数就好了。

由于数字范围只有1~K，，，那就直接开个桶。

被坑了一发!

不知道为什么，先是这样写：

ans=ans-times[a[x]]*times[a[x]]+(++times[a[x]])*times[a[x]];

感觉没有问题啊。。可是WA了好久，，然后改成了：

ans-=times[a[x]]*times[a[x]];
times[a[x]]++;
ans+=times[a[x]]*times[a[x]];

瞬间AC…………

不说什么了……QAQ

#include<bits/stdc++.h>
#define ll long long
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    while (ch<'0' || ch>'9'){if (ch=='-') f=-1;ch=getchar();}
    while (ch>='0' && ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
const int 
	Block=230,
	MAX=50005;
int n,m,k,a[MAX];
ll ans,Ans[MAX],times[MAX];
struct Query{
	int L,R,id;
}Q[MAX];
bool cmp(Query x,Query y){
	if ((x.L/Block)!=(y.L/Block))
		return x.L<y.L;
	return x.R<y.R;
}
void add(int x){
	ans-=times[a[x]]*times[a[x]];
	times[a[x]]++;
	ans+=times[a[x]]*times[a[x]];
}
void remove(int x){
	ans-=times[a[x]]*times[a[x]];
	times[a[x]]--;
	ans+=times[a[x]]*times[a[x]];
}
int main(){
	n=read(),m=read(),k=read();
	for (int i=1;i<=n;i++) a[i]=read();
	for (int i=1;i<=m;i++)
		Q[i].L=read(),Q[i].R=read(),Q[i].id=i;
	sort(Q+1,Q+1+m,cmp);
	memset(times,0,sizeof(times));
	int l=1,r=0; ans=(ll)0;
	for (int i=1;i<=m;i++){
		while (l<Q[i].L) remove(l++);
		while (l>Q[i].L) add(--l);
		while (r<Q[i].R) add(++r);
		while (r>Q[i].R) remove(r--);
		Ans[Q[i].id]=ans;
	}
	for (int i=1;i<=m;i++)
		printf("%lld\n",Ans[i]);
	return 0;
}