luogu P2709 小B的询问 最简单的莫队

块内按右端点sort，块外按左端点sort

话说我刚开始这么修改。。。

inline void del(int i) {cnt-=(c[a[i]]*c[a[i]]),cnt+=(c[a[i]]-)*(c[a[i]]-),--c[a[i]];}
inline void ins(int i) {cnt-=(c[a[i]]*c[a[i]]),cnt+=(c[a[i]]+)*(c[a[i]]+),++c[a[i]];}

十分暴力？没事不影响复杂度。。

后来看题解发现可以改成这样

inline void del(int i) {--c[a[i]],cnt-=*c[a[i]]+;}
inline void ins(int i) {++c[a[i]],cnt+=*c[a[i]]-;}

十分简练？我数学太菜了。。。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#define R register int
using namespace std;
inline int g() {
    R ret=,fix=; register char ch; while(!isdigit(ch=getchar())) fix=ch=='-'?-:fix;
    do ret=ret*+(ch^); while(isdigit(ch=getchar())); return ret*fix;
}
int n,m,k,T,l=,r=;
long long cnt=,ans[];
int c[],a[];
struct seg {
    int l,r,rk;
    bool operator <(const seg& y)const{return (l-)/T==(y.l-)/T?r<y.r:l<y.l;}
}q[];
inline void del(int i) {--c[a[i]],cnt-=*c[a[i]]+;}
inline void ins(int i) {++c[a[i]],cnt+=*c[a[i]]-;}
signed main() {
    n=g(),m=g(),k=g(); T=1.2*sqrt(n);
    for(R i=;i<=n;++i) a[i]=g();
    for(R i=;i<=m;++i) q[i].l=g(),q[i].r=g(),q[i].rk=i;
    sort(q+,q+m+);
    for(R i=;i<=m;++i) {
        while(l<q[i].l) del(l++);
        while(l>q[i].l) ins(--l);
        while(r<q[i].r) ins(++r);
        while(r>q[i].r) del(r--);
        ans[q[i].rk]=cnt;
    } for(R i=;i<=m;++i) printf("%lld\n",ans[i]);
}

---

2019.04.12