P2709 小B的询问

Posted 2020-12-05 lovedsr

• 题目描述

  小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

普通莫队没啥好说的

类似于 小z的袜子

也是类似的维护平方

感觉莫队这么容易上手233

//begin 16.50
//end  17.35
#include <cmath>
#include <iostream>
#include <algorithm>
#include <cstdio>
#define maxn 50010
using namespace std;
inline int read()
{
    int x=0,f=1;char s=getchar();
    while('0'>s||s>'9'){if(s=='-')f=-1;s=getchar();}
    while('0'<=s&&s<='9'){x=x*10+s-'0';s=getchar();}
    return x*f;
}
int n,m,k,now;
int belong[maxn];
struct node{
    int x,y,id;
}q[maxn];
bool operator < (const node &a,const node &b)
{
    return belong[a.x]==belong[b.x] ? a.y<b.y : a.x<b.x;
}
int a[maxn];//原数 
int c[maxn];//出现的次数 
int ans[maxn];
void Add(int x)
{
    now-=c[x]*c[x];
    ++c[x];
    now+=c[x]*c[x];
}
void Delet(int x)
{
    now-=c[x]*c[x];
    --c[x];
    now+=c[x]*c[x];
}
int main()
{
    n=read();
    m=read();
    k=read();
    int kk=sqrt(n);
    for(int i=1;i<=n;++i)
        a[i]=read();
    for(int i=1;i<=n;++i)
        belong[i]=(i-1)/kk+1;
    for(int i=1;i<=m;++i)
    {
        q[i].x=read();
        q[i].y=read();
        q[i].id=i;
    }
    sort(q+1,q+1+m);
    int l=1,r=0;
    for(int i=1;i<=m;++i)
    {
        //cout<<"Debug:
"<<l<<" "<<r<<" ->"<<q[i].x<<" "<<q[i].y<<"
";
        while(l > q[i].x) Add(a[--l]);
        while(l < q[i].x) Delet(a[l++]);
        while(r > q[i].y) Delet(a[r--]);
        while(r < q[i].y) Add(a[++r]);
        ans[q[i].id]=now;
    }
    for(int i=1;i<=m;++i)
        printf("%d
",ans[i]);
    return 0;
}