bzoj3781 小B的询问

题目

莫队第一题，照样先不理解，写板子。

记板，记板。

虽然不太典型，但也练练吧。

思路最重要。

//莫队算法 
#include<bits/stdc++.h>
#define N 50000
using namespace std;
struct que{
    int l;
    int r;
    int block;
    int id;
    bool operator < (const que &A) const
    {
        if(block==A.block)return r<A.r;
        return block<A.block;
    }
};que A[N+5];
int n,m,k,siz,seq[N+1],l,r; 
long long cnt[N+1],ans,res[N+1];
inline char nc()
{
    static char buf[100000],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int read()
{
    int x=0,b=1;
    char c=nc();
    for(;!(c<='9'&&c>='0');c=nc())if(c=='-')b=-1;
    for(;c<='9'&&c>='0';c=nc())x=x*10+c-'0';
    return x*b;
}
inline void write(long long x)
{
    if(x==0)putchar('0');
    else
    {
        char buf[25];
        int len=0;
        if(x<0)putchar('-'),x=-x;
        while(x)buf[++len]=x%10+'0',x/=10;
        for(int i=len;i>=1;i--)putchar(buf[i]);
    }
    putchar('\n');
}
int main()
{
    freopen("in.txt","r",stdin);
    n=read();m=read(),k=read();
    siz=sqrt(n);
    for(int i=1;i<=n;i++)seq[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)/siz+1;
    sort(A+1,A+m+1);
    l=1,r=0,ans=0;
    for(int i=1;i<=m;i++)
    {
        while(l>A[i].l)l--,cnt[seq[l]]++,ans+=2*cnt[seq[l]]-1;
        while(r<A[i].r)r++,cnt[seq[r]]++,ans+=2*cnt[seq[r]]-1;
        while(l<A[i].l)cnt[seq[l]]--,ans-=2*cnt[seq[l]]+1,l++;
        while(r>A[i].r)cnt[seq[r]]--,ans-=2*cnt[seq[r]]+1,r--;
        res[A[i].id]=ans;
    }
    for(int i=1;i<=m;i++)write(res[i]);
    return 0;
}