#莫队二次离线,根号分治#洛谷 5398 [Ynoi2018] GOSICK

题目

\(m\) 组询问求 \(\sum_{l\leq i,j\leq r}[a_i\bmod a_j==0],n,m,a_i\leq 5\times 10^5\)


分析

设 \(f(l,r,x)\) 表示 \(i 或 j\in [1,x],i或j\in [l,r]\) 时 的答案,\(g_x\) 表示 \([1,x]\) 的答案,根号的做法可以通过三秒

由于涉及区间内的求值,需要在莫队的基础上二次离线,那么从 \([l,r]\) 的答案扩展到 \([l,r']\) 就要加上 \(g(r')-g(r)-f(r+1,r',l-1)\)

从 \([l,r]\) 扩展到 \([l',r]\) 就要加上 \(f(l',l-1,r)-[g(l-1)-g(l'-1)]\),现在就是要求 f 和 g,后者可以通过前缀和实现

如果 \([l,r]\) 是作为因数,那么直接将 \(a_{1\sim x}\) 的因数统计个数,以 \(x\) 为第一关键字,即可做到 \(O(m\sqrt{n})\)

如果作为倍数,那么 \(a_{1\sim x}\) 中大于阈值的直接枚举倍数统计个数,小于等于的就要单独抽离出来,倘若现在枚举到 \(z\)

设 \(c_i\) 表示前 \(i\) 个数中 \(z\) 的倍数的个数,那么区间的答案就可以转化成前 \(x\) 个数中 \(z\) 的个数乘上 \(c_r-c_{l-1}\)

然后莫队的块长调到 \(1.5\sqrt{n}\),阈值调到 \(0.15\sqrt{\max\{a_i\}}\) 就可以过了,注意求 \(g_i\) 的时候要给每个位置减一,

因为 \((i,i)\) 和 \((i,i)\) 等价,但 \((i,j)\) 和 \((j,i)\) 在 \(a_i=a_j\) 时不等价


代码

cpp 复制代码
#include <cstdio>
#include <cctype>
#include <cmath>
#include <algorithm>
#define getchar() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin)),p1==p2?EOF:*p1++)
using namespace std;
const int N=500011; char buf[1<<21],puf[1<<21],*p1,*p2; int nowp=-1;
struct rec{int l,r,rk;}q[N];
struct node{int y,next;}e[N*27];
struct Node{int x,y,rk,z,next;}E[N<<1];
typedef long long lll; lll s[N],ans[N];
int n,Q,bl,c[N],pos[N],a[N],b[N],as[N],bs[N],hs[N],rk[N],et,Et,m;
int iut(){
    int ans=0; char c=getchar();
    while (!isdigit(c)) c=getchar();
    while (isdigit(c)) ans=ans*10+c-48,c=getchar();
    return ans;
}
void Flush(){fwrite(puf,1,nowp+1,stdout),nowp=-1;}
void Putchar(char x){
	if (nowp==(1<<21)) Flush();
	puf[++nowp]=x;
}
void print(lll ans){
	char dig[21]; int len=-1;
	do{
		dig[++len]=ans%10+48,ans/=10;
	}while (ans);
	while (len>=0) Putchar(dig[len--]);
}
bool cmp(rec x,rec y){
    if (pos[x.l]!=pos[y.l]) return pos[x.l]<pos[y.l];
    return (pos[x.l]&1)?(x.r<y.r):(x.r>y.r);
}
int main(){
    n=iut(),Q=iut(),bl=sqrt(n)*1.5;
    for (int i=1;i<=n;++i) b[i]=a[i]=iut(),pos[i]=(i-1)/bl+1;
    for (int i=1;i<=Q;++i) q[i]=(rec){iut(),iut(),i};
    sort(q+1,q+1+Q,cmp);
    sort(b+1,b+1+n),m=unique(b+1,b+1+n)-b-1;
	bl=sqrt(m)*0.15+1;
    for (int i=1;i<=m;++i) rk[b[i]]=i;
    for (int i=1;i<=n;++i) a[i]=rk[a[i]];
    for (int i=1;i<=m;++i)
    for (int j=b[i];j<=b[m];j+=b[i])
    if (rk[j]){
        e[++et]=(node){rk[j],as[i]},as[i]=et;
        e[++et]=(node){i,bs[rk[j]]},bs[rk[j]]=et;
    }
    int L=q[1].l,R=q[1].l-1;
    for (int i=1;i<=Q;++i){
        if (L>q[i].l) E[++Et]=(Node){q[i].l,L-1,q[i].rk,1,hs[R]},hs[R]=Et,L=q[i].l;
        if (L<q[i].l) E[++Et]=(Node){L,q[i].l-1,q[i].rk,-1,hs[R]},hs[R]=Et,L=q[i].l;
        if (R>q[i].r) E[++Et]=(Node){q[i].r+1,R,q[i].rk,1,hs[L-1]},hs[L-1]=Et,R=q[i].r;
        if (R<q[i].r) E[++Et]=(Node){R+1,q[i].r,q[i].rk,-1,hs[L-1]},hs[L-1]=Et,R=q[i].r;
    }
    for (int i=1;i<=bl;++i){
        for (int j=1;j<=n;++j)
        if (b[a[j]]%b[i]==0) c[j]=c[j-1]+1;
            else c[j]=c[j-1];
		int now=0;
        for (int j=1;j<=n;++j){
            if (a[j]==i) ++now;
            if (b[a[j]]%b[i]==0) s[j]+=now;
            for (int k=hs[j];k;k=E[k].next)
                ans[E[k].rk]+=(c[E[k].y]-c[E[k].x-1])*now*E[k].z;
        }
    }
    for (int i=1;i<=n;++i) c[i]=0;
    for (int i=1;i<=n;++i){
        for (int j=bs[a[i]];j;j=e[j].next) ++c[e[j].y];
        if (a[i]>bl) for (int j=as[a[i]];j;j=e[j].next) ++c[e[j].y];
        s[i]+=c[a[i]]-1;
        for (int j=hs[i];j;j=E[j].next){
        	int now=0;
        	for (int k=E[j].x;k<=E[j].y;++k)
        	    now+=c[a[k]];
        	ans[E[j].rk]+=now*E[j].z;
		}
    }
    for (int i=1;i<=n;++i) s[i]+=s[i-1];
    for (int i=1;i<=Et;++i) ans[E[i].rk]-=E[i].z*(s[E[i].y]-s[E[i].x-1]);
    for (int i=1;i<=Q;++i) ans[q[i].rk]+=ans[q[i-1].rk];
    for (int i=1;i<=Q;++i) print(ans[i]),Putchar(10);
    Flush();
    return 0;
}