[BZOJ3781] 小B的询问

题目描述

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<=NMK<=50000

题目分析:

BZOJ权限题 具体题目描述请看洛谷
莫队第一题...... 具体做法转到:莫队讲解 + 莫队讲解

#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <algorithm>
using namespace std;
int n,m,k;
int c[60000];
struct your
{
	int x,y,id,block;
}a[60000];
int siz;
int ans[60000],vis[600000];
bool cmp(your j,your k)
{
	if(j.block==k.block) return j.y<k.y;
	return j.block<k.block;
}
int sum;
void add(int x)
{
	sum+=2*vis[c[x]]+1;
	vis[c[x]]++;
}
void sub(int x)
{
	sum-=2*vis[c[x]]-1;
	vis[c[x]]--;
}
int main()
{
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;i++) scanf("%d",&c[i]);
	siz=sqrt(n);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&a[i].x,&a[i].y);
		a[i].id=i;
		a[i].block=(a[i].x-1)/siz+1;
	}
	sort(a+1,a+m+1,cmp);
	int l=1,r=0;
	for(int i=1;i<=m;i++)
	{
		while(l<a[i].x) sub(l++); 
		while(l>a[i].x) add(--l); 
		while(r<a[i].y) add(++r);
		while(r>a[i].y) sub(r--);
		ans[a[i].id]=sum;
	}
	for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
 return 0;
}

发表评论

邮箱地址不会被公开。 必填项已用*标注