51nod 1742 开心的小Q

原题链接：
https://www.51nod.com/onlineJudge/questionCode.html#problemId=1742&noticeId=324902

$$answer=\sum_{i=1}^n\sum_{d|i}\big(1-\mu^2(i)\big)\\=\sum_{i=1}^n\sum_{d|i}1-\sum_{i=1}^n\sum_{d|i}^n\mu^2(i)\\=\sum_{d=1}^n\sum_{i\leq n,d|i}1-\sum_{d=1}^n\sum_{i\leq n,d|i}\mu^2(i)\\=\sum_{d=1}^n\big\lfloor\frac{n}{d}\big\rfloor-\sum_{d=1}^n\mu^2(d)\big\lfloor\frac{n}{d}\big\rfloor$$

对于 $$\sum_{i=1}^n\mu^2(i)$$

定义：$f(k)$为$[1,n]$中，平方因子最大为$k^2$的集合。

比如：$32$的因子有：$1,2,4,8,16,32$

其中，最大的平方因子是$16$.

那么对于一个数字。他拆成这种表示唯一的。（形如$a^2b$）

定义： $$F(k)=\sum_{k|d}f(d)=\big\lfloor\frac{n}{k^2}\big\rfloor$$

$$\sum_{i=1}^n\mu^2(i)=f(1)=\sum_{k=1}^n\mu(k)F(k)=\sum_{k=1}^n\mu^2(k)\big\lfloor\frac{n}{k^2}\big\rfloor$$

即：

$$\sum_{k=1}^{\lfloor\sqrt{n}\rfloor}\mu^2(k)\big\lfloor\frac{n}{k^2}\big\rfloor$$

分段计算即可。

#include <algorithm>
#include <string.h>
#include <cmath>
#define MAXN 1111111
using namespace std;
typedef long long LL;
const int m=1111;
int mu[MAXN]={0,-1};
int M(int n)
{
    int ans=0;
    for(int i=1;i*i<=n;i++) ans+=mu[i]*(n/(i*i));
    return ans;
}
LL slove(int n)
{
    if(n==0)return 0;
    LL ans=0;
    if(n<MAXN)
        for(int i=1;i<=n;i++)   ans+=(mu[i])?0:(n/i);
    else
    {
        int bu=M(n);
        for(int L=1;L<m;L++)
        {
            int k=M(n/(L+1));
            ans-=(LL)L*(bu-k);
            bu=k;
        }
        for(int i=n/m;i;i--)    ans-=(mu[i])?(n/i):0;
        int lim=sqrt(n)+1;
        for(int L=1;L<lim;L++) ans+=(LL)L*(n/L-n/(L+1));
        for(int i=n/lim;i;i--) ans+=n/i;
    }
    return ans;
}
int main ()
{
    for(int i=1;i<MAXN;i++)
    {
        mu[i]=-mu[i];
        for(int j=i<<1;j<MAXN;j+=i) mu[j]+=mu[i];
    }
    int a,b;
    scanf("%d %d",&a,&b);
    printf("%lld\n",slove(b)-slove(a-1));
    return 0;
}