莫比乌斯反演 - Visible Lattice Points - SPOJ - VLATTICE

莫比乌斯反演 - Visible Lattice Points - SPOJ - VLATTICE

题意：

$给定一个n\times n\times n的立方体，从(0,0,0)看过去，问能够看到多少个点。$

$空间中，一个点X能够被点Y看到的充要条件是：点X和点Y的这条线段上，不存在其他点。$

Input :

The first line contains the number of test cases T. The next T lines contain an interger n

Output :

Output T lines, one corresponding to each test case.

Sample Input :

3
1
2
5

Sample Output :

7
19
175

Constraints :

T <= 50
1 <= N <= 1000000

---

分析：

$我们知道，在二维坐标系中，点(x_1,y_1)和点(x_2,y_2)之间不存在其他点的充要条件是:$

$$gcd(x_2-x_1,y_2-y_1)=1$$

$在三维坐标系中，点(x_1,y_1,z_1)和点(x_2,y_2,z_2)之间不存在其他点的充要条件是:$

$$gcd(x_2-x_1,y_2-y_1,z_2-z_1)=1$$

$由于我们从(0,0,0)看，所以条件为：gcd(x_2-0,y_2-0,z_2-0)=1$

$由于三维坐标(i,j,k)都是可能为0的，因此我们先考虑特殊情况：$

$①、当其中有两维坐标为0时，我们能够看到3个点：(1,0,0)、(0,1,0)、(0,0,1)$

$②、当其中有一维坐标为0时，能够看到的点的数量是：3\times {\sum }_{i=1}^n{\sum }_{j=1}^n[gcd(i,j)==1]$

$③、当所有坐标都不为0时，能够看到的点点数量是：{\sum }_{i=1}^n{\sum }_{j=1}^n{\sum }_{k=1}^n[gcd(i,j,k)==1]$

$则答案为：$

$$Ans=3+3\times \sum _{i=1}^n\sum _{j=1}^n[gcd(i,j)==1]+\sum _{i=1}^n\sum _{j=1}^n\sum _{k=1}^n[gcd(i,j,k)==1]$$

$根据反演定理，我们很快得到：$

$$\sum _{i=1}^n\sum _{j=1}^n[gcd(i,j)==1]=\sum _{i=1}^n\mu (i)\lfloor \frac{n}{i}\rfloor \lfloor \frac{n}{i}\rfloor$$

$同理:$

$$\sum _{i=1}^n\sum _{j=1}^n\sum _{k=1}^n[gcd(i,j,k)==1]=\sum _{i=1}^n\mu (i)\lfloor \frac{n}{i}\rfloor \lfloor \frac{n}{i}\rfloor \lfloor \frac{n}{i}\rfloor$$

$综上答案为：$

$$Ans=3+\sum _{i=1}^n\mu (i)\lfloor \frac{n}{i}\rfloor \lfloor \frac{n}{i}\rfloor (\lfloor \frac{n}{i}\rfloor +3)$$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>

#define ll long long

using namespace std;

const int N = 1000010;

int T, n;
int mu[N], primes[N], cnt, sum[N];
bool st[N];
ll f[N];

void get_prime(int n)
{
    mu[1]=1;
    for(int i=2;i<=n;i++)
    {
        if(!st[i])
        {
            primes[cnt++]=i;
            mu[i]=-1;
        }
        for(int j=0;primes[j]*i<=n;j++)
        {
            st[primes[j]*i]=true;
            if(i%primes[j]==0)
            {
                mu[i*primes[j]]=0;
                break;
            }
            mu[i*primes[j]]=mu[i]*-1;
        }
    }
    for(int i=1;i<=n;i++) sum[i]=sum[i-1]+mu[i];
}

ll cal(int n)   
{
    ll res=0;
    for(int i=1, r=1;i<=n;i=r+1)
    {
        r=n/(n/i);
        r=min(r,n);
        res+=(ll)(sum[r]-sum[i-1])*(n/i)*(n/i)*(n/i+3);
    }
    return res;
}

int main()
{
    get_prime(N-1);
    memset(f,-1,sizeof f);
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        printf("%lld\n",cal(n)+3);
    }
    
    return 0;
}