BZOJ3701 Olympic Games 莫比乌斯反演

题目描述

• 给出$n,m,l,r,mod$
• 求$(n+1)\ast (m+1)$的网格上能相互看见的点对数对mod取模的值
• 能相互看见定义为两点的连线不经过其他点，且欧氏距离在[l,r]范围内
• $n,m\le 100000$
  $l,r\le 150000$
  $mod\le 10^9$

题目分析

粗略描述一下题目要求:
$$ans=\sum _{0\le i\le n}\sum _{0\le j\le m}\sum _{0\le x\le n}\sum _{0\le y\le n}[(i-x,j-y)=1][l\le \sqrt{(i-x{)}^2+(j-y{)}^2}\le r]$$

显然过于繁琐，考虑直接看$i-x$和$j-y$，即枚举两个坐标的差值
($PS:$ 对相邻两点单独处理，即当$l\le 1\le r时，ans+=2\ast n\ast m+n+m$)

由于距离大小都有限制，所以转化为求距离<=r的减去距离<=l-1的

$$Ans(L^2)=\sum _{1\le i\le n}2\ast (n-i+1)\sum _{1\le j\le min(m,\sqrt{L^2-i^2})}[(i,j)=1]\ast (m-j+1)$$
莫比乌斯反演换掉条件框，
$$Ans(L^2)=\sum _{1\le i\le n}2\ast (n-i+1)\sum _{1\le j\le min(m,\sqrt{L^2-i^2})}\sum _{d|(i,j)}\mu (d)\ast (m-j+1)$$
$$=\sum _{1\le i\le n}2\ast (n-i+1)\sum _{d|i}\mu (d)\sum _{j=1}^{min(m/d,\sqrt{L^2-i^2}/d)}(m-dj+1)$$
枚举$i,d,O(n\sqrt{n})$，后面那坨可以$O(1)$求

当初做的时候没有想到可以直接枚举，到最后一步的时候想着要把d提到最前面，然后按照套路把i,j转成倍数形式，结果懵逼好久，所以该枚举的时候就要大胆枚举，要为了解题而用莫比乌斯。。。
注意到处是坑的$1ll$…

#include<cstdio>
#include<cmath>
#include<algorithm>
#define maxn 100005
#define LL long long
using namespace std;
int n,m,mod,l,r,mu[maxn],p[maxn];
bool v[maxn];
void Prime(int N)
{
    mu[1]=1;int cnt=0;
    for(int i=2;i<=N;i++)
    {
        if(!v[i]) p[++cnt]=i,mu[i]=-1;
        for(int j=1,k;j<=cnt&&p[j]*i<=N;j++)
        {
            v[k=p[j]*i]=1;
            if(i%p[j]==0) {mu[k]=0;break;}
            mu[k]=-mu[i];
        }
    }
}
inline int calc(int d,int j){return mu[d]*(1ll*j*m-1ll*d*j*(j+1)/2+j)%mod;}
LL solve(LL R)
{
    LL ret=0;
    for(int i=1;i<=n&&1ll*i*i<R;i++)
        for(int d=1,j;d*d<=i;d++)
            if(i%d==0)
            {
                j=min(int(sqrt(R-1ll*i*i)/d),m/d);
                ret=(ret+1ll*(n-i+1)*calc(d,j))%mod;
                if(i/d!=d){
                    j=min(int(sqrt(R-1ll*i*i)/(i/d)),m/(i/d));
                    ret=(ret+1ll*(n-i+1)*calc(i/d,j))%mod;
                }
            }
    return ret*2%mod;
}
int main()
{
    scanf("%d%d%d%d%d",&n,&m,&l,&r,&mod);
    Prime(n);
    LL ans=solve(1ll*r*r)-solve(1ll*l*l-1);
    if(l<=1&&1<=r) ans+=1ll*n*m*2+n+m;
    printf("%lld",(ans%mod+mod)%mod);
}