bzoj2820 YY的GCD

题目

这肯定要用卷积的，我们来推一推公式。

我们先令$n=min(n,m)$,$m=max(m,n)$

则
$ans=\sum^{}_{p}\sum^{n}_{i=1}\sum^{m}_{j=1}gcd(i,j)==p$
$ans=\sum^{}_{p}\sum^{[\frac{n}{p}]}_{i=1}\sum^{[\frac{m}{p}]}_{j=1}gcd(i,j)==1$
$ans=\sum^{}_{p}\sum^{[\frac{n}{p}]}_{i=1}\sum^{[\frac{m}{p}]}_{j=1}\sum_{d|gcd(i,j)}\mu(d)$
$ans=\sum^{}_{p}\sum^{[\frac{n}{p}]}_{d=1}\mu(d)[\frac{n}{pd}][\frac{m}{pd}]$
现在，我们就可以在素数个数乘以$\sqrt{n}$下完成一个询问了，但是，这样显然会超时。
继续化简式子，令$k=pd$
则
$ans=\sum^{n}_{k=1}\sum_{p|k}\mu(\frac{k}{p})[\frac{n}{k}][\frac{m}{k}]$
再令$F(x)=\sum_{p|x}\mu(\frac{x}{p})$，这个东西可以$n\log{n}算出来的$

最后就可以$\sqrt{n}$完成一次询问了，这道题就解决了。

#include<bits/stdc++.h>
#define N 10000000
using namespace std;
typedef long long ll; 
int prime[N+5],notp[N+5],mu[N+5],cnt,T,n,m,lst;
ll F[N+5],ans;
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 init()
{
    notp[1]=true,mu[1]=1;
    for(int i=2;i<=N;i++)
    {
        if(!notp[i])prime[++cnt]=i,mu[i]=-1;
        for(int j=1;j<=cnt&&i*prime[j]<=N;j++)
        {
            notp[i*prime[j]]=true;
            if(i%prime[j]==0)
            {
                mu[i*prime[j]]=0;
                break;
            }
            mu[i*prime[j]]=-mu[i];
        }
    }
    for(int i=1;i<=cnt;i++)
    {
        int p=prime[i];
        for(int j=1;j*p<=N;j++)F[j*p]+=mu[j];
    }
    for(int i=1;i<=N;i++)F[i]+=F[i-1];
}
int main()
{
    freopen("in.txt","r",stdin);
    init();
    T=read();
    while(T--)
    {
        n=read(),m=read();if(n>m)swap(n,m);ans=0;
        for(int i=1;i<=n;i=lst+1)
        {
            lst=min(n/(n/i),m/(m/i));
            ans+=(ll)(F[lst]-F[i-1])*(n/i)*(m/i);
        }
        printf("%lld\n",ans);
    }
    return 0;
}