洛谷 P2709 小B的询问（莫队）

题目链接：https://www.luogu.com.cn/problem/P2709

这道题是模板莫队，然后$i$在$[l,r]$区间内的个数就是$vis[ ]$数组

$add()$和$del()$的话就是先减去原来位置数的个数的平方，然后再加上现在位置数的个数的平方。

AC代码：

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

using namespace std;
const int N=50005;

int a[N],vis[N],ans,tot[N];
int block;

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

bool cmp(node aa,node bb){
    if(aa.l/block==bb.l/block) return aa.r<bb.r;
    return aa.l/block<bb.l/block;
}

void add(int pos){
    ans-=vis[a[pos]]*vis[a[pos]];
    vis[a[pos]]++;
    ans+=vis[a[pos]]*vis[a[pos]];
}

void del(int pos){
    ans-=vis[a[pos]]*vis[a[pos]];
    vis[a[pos]]--;
    ans+=vis[a[pos]]*vis[a[pos]];
}

int main(){
    int n,m,k;
    scanf("%d%d%d",&n,&m,&k);
    block=sqrt(n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&q[i].l,&q[i].r);
        q[i].id=i;
    }
    sort(q+1,q+m+1,cmp);
    int L=1,R=0;
    for(int i=1;i<=m;i++){
        while(L>q[i].l){
            L--;
            add(L);
        }
        while(L<q[i].l){
            del(L);
            L++;
        }
        while(R<q[i].r){
            R++;
            add(R);
        }
        while(R>q[i].r){
            del(R);
            R--;
        }
        tot[q[i].id]=ans;
    }
    for(int i=1;i<=m;i++) printf("%d
",tot[i]);
    return 0;
}