洛谷 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

分析：显然可以用莫队乱搞。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>

const int maxn=50005;

using namespace std;

struct node{
    int id,l,r,ans;
}q[maxn];

int a[maxn],sum[maxn],belong[maxn];
int n,m,i,k,block,l,r,ans;

bool cmp(node x,node y)
{
    if (belong[x.l]==belong[y.l]) return x.r<y.r;
    return x.l<y.l;
}

bool cmp1(node x,node y)
{
    return x.id<y.id;
}

void updata(int x,int c)
{
    if (c==1)
    {
        ans+=sum[a[x]]*2+1;
        sum[a[x]]++;
    }
    else
    {
        ans-=sum[a[x]]*2-1;
        sum[a[x]]--;
    }
}

int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for (i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
    }
    for (i=1;i<=m;i++)
    {
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    block=trunc(sqrt(n));
    for (i=1;i<=n;i++) belong[i]=(i-1)/block+1; 
    sort(q+1,q+m+1,cmp);
    l=1; r=1; sum[a[1]]=1; ans=1;
    for (i=1;i<=m;i++)
    {
        for (r;r<q[i].r;r++) updata(r+1,1);
        for (r;r>q[i].r;r--) updata(r,0);
        for (l;l<q[i].l;l++) updata(l,0);
        for (l;l>q[i].l;l--) updata(l-1,1);
        q[i].ans=ans;
    }
    sort(q+1,q+1+m,cmp1);
    for (i=1;i<=m;i++) printf("%d\n",q[i].ans);
}