bzoj3781: 小B的询问

莫队，队询问排序，以左端点块为第一关键字，右端点为第二关键字，离线处理。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define ll long long
using namespace std;
int read()
{
    char ch=getchar();int f=0;
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9'){f=(f<<1)+(f<<3)+ch-'0';ch=getchar();}
    return f;
}
int n,m,k;
int b[50005];
ll cnt[50005],res[50005];
struct query
{
    int l,r,block,id;
}a[50005];
bool operator <(query a,query b)
{
    if(a.block==b.block) return a.r<b.r;
    return a.block<b.block;
}
void solve()
{
    sort(a+1,a+m+1);
    int l=1,r=0;ll ans=0;
    for(int i=1;i<=m;i++)
    {
        while(l>a[i].l) l--,cnt[b[l]]++,ans+=2*cnt[b[l]]-1;
        while(r<a[i].r) r++,cnt[b[r]]++,ans+=2*cnt[b[r]]-1;
        while(l<a[i].l) cnt[b[l]]--,ans-=2*cnt[b[l]]+1,l++;
        while(r>a[i].r) cnt[b[r]]--,ans-=2*cnt[b[r]]+1,r--;
        res[a[i].id]=ans;
    }
}
int main()
{
    n=read();m=read(),k=read();
    int size=sqrt(n);
    for(int i=1;i<=n;i++) b[i]=read();
    for(int i=1;i<=m;i++)
    {
        a[i].l=read(),a[i].r=read();
        a[i].id=i;
        a[i].block=(a[i].l-1)/size+1;
    }
    solve();
    for(int i=1;i<=m;i++) printf("%lld\n",res[i]);
    return 0;
}