【反演复习计划】【bzoj1011】zap-queries

快三个月没做反演题了吧……

感觉高一上学期学的全忘了……

所以还得从零开始学推式子。

# bzoj1011

标签（空格分隔）： 未分类

---

原题意思是求以下式子：
$Ans=\sum\limits_{i=1}^{a}\sum\limits_{i=1}^{b}[gcd(i,j)==k]$
首先把k拿下来，得到
$Ans=\sum\limits_{i=1}^{a/k}\sum\limits_{i=1}^{b/k}[gcd(i,j)==1]$
然后考虑mobius函数的性质：
$\sum\limits_{d|n}\mu(d)=1(n==1),0(n>1)$
所以可以把那个gcd的式子替换下，得到：
$Ans=\sum\limits_{i=1}^{a/k}\sum\limits_{i=1}^{b/k}\sum\limits_{d|gcd(i,j)}\mu(d)$
我们稍微改写一下这个式子：
$Ans=\sum\limits_{i=1}^{a/k}\sum\limits_{i=1}^{b/k}\sum\limits_{d|i,d|j}\mu(d)$
这个时候我们把$\mu(i)$提前（也就是交换枚举顺序）得到下面的式子：
$Ans=\sum\limits_{d=1}^{min(a/k,b/k)}\mu(d)\sum\limits_{i=1,d|i}^{a/k}\sum\limits_{j=1,d|j}^{b/k}1$
这个式子比较蠢，我们能看出来这个式子的意思就是：
$Ans=\sum\limits_{d=1}^{min(a/k,b/k)}\mu(d)\frac{a/k}{d}\frac{b/k}{d}$
考虑到后者只有$\sqrt{\frac{a}{k}}$种取值
所以下底函数分块，前缀和优化下就能过了。

#include<bits/stdc++.h>
#define N 100005
using namespace std;
typedef long long ll;
int prime[N],mu[N],s[N],vis[N],cnt=;
void calcmu(){
    cnt=;mu[]=;memset(vis,,sizeof(vis));
    for(int i=;i<N;i++){
        if(vis[i])prime[++cnt]=i,mu[i]=-;
        for(int j=;j<=cnt;j++){
            int t=prime[j]*i;if(t>N)break;
            vis[t]=;
            if(i%prime[j]==){mu[t]=;break;}
            mu[t]=-mu[i];
        }
    }
    s[]=;
    for(int i=;i<=N;i++)s[i]=s[i-]+mu[i];
}
ll calc(int n,int m,int k){
    n/=k;m/=k;ll ans=;int j=;
    if(n>m)swap(n,m);
    for(int i=;i<=n;i=j+){
        j=min(n/(n/i),m/(m/i));
        ans+=1LL*(s[j]-s[i-])*(n/i)*(m/i);
    }
    return ans;
}
inline int read(){
    int f=,x=;char ch;
    do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
    do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
    return f*x;
}
int main(){
    int T=read();calcmu();
    while(T--){
        int n=read(),m=read(),k=read();
        printf("%lld\n",calc(n,m,k));
    }
    return ;
}