P2709 小B的询问

题目描述

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

输入输出格式

输入格式：

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

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

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

输出格式：

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

输入输出样例

输入样例#1： 

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

输出样例#1： 

6
9
5
2

说明

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

Solution：

　　本题是一道莫队模板题。

　 直接由题意，维护每种颜色数的平方，然后就是莫队的分块->块内排序->指针移动->记录答案。

　 注意一下排序时先判断左端所在块是否相同，相同就按右段升序排列，否则就按左端升序排列。

　 本题就这么简单的$AC$了

代码：

#include<bits/stdc++.h>
#define il inline
#define ll long long
using namespace std;
const int N=100005;
int n,m,k,pos[N],a[N],s[N],ans[N],tot;
struct data{
    int l,r,id;
}t[N];
il int gi()
{
    int a=0;char x=getchar();bool f=0;
    while((x<'0'||x>'9')&&x!='-')x=getchar();
    if(x=='-')x=getchar(),f=1;
    while(x>='0'&&x<='9')a=a*10+x-48,x=getchar();
    return f?-a:a;
}
il bool cmp(data a,data b){return pos[a.l]==pos[b.l]?a.r<b.r:a.l<b.l;}
il void change(int k,int u)
{
    tot-=s[a[k]]*s[a[k]];
    s[a[k]]+=u;
    tot+=s[a[k]]*s[a[k]];
}
int main()
{
    n=gi(),m=gi(),k=gi();
    int ss=int(sqrt(n));
    for(int i=1;i<=n;i++)a[i]=gi(),pos[i]=(i-1)/ss+1;
    for(int i=1;i<=m;i++)t[i].l=gi(),t[i].r=gi(),t[i].id=i;
    sort(t+1,t+m+1,cmp);
    for(int i=1,l=1,r=0;i<=m;i++){
        while(r<t[i].r)change(r+1,1),r++;
        while(r>t[i].r)change(r,-1),r--;
        while(l>t[i].l)change(l-1,1),l--;
        while(l<t[i].l)change(l,-1),l++;
        ans[t[i].id]=tot;
    }
    for(int i=1;i<=m;i++)printf("%d
",ans[i]);
    return 0;
}