JZOJ 4919. 【NOIP2017提高组模拟12.10】神炎皇

Description

神炎皇乌利亚很喜欢数对，他想找到神奇的数对。

对于一个整数对(a,b)，若满足a+b<=n且a+b是ab的因子，则成为神奇的数对。请问这样的数对共有多少呢？

Solution

我们设一个数对(a,b)的最大公因数为d，a’*d=a,b’ *d=b,gcd(a’,b’)=1,那式子一定满足$d(a'+b')|d^2a'b'$。因为gcd(a’,b’)=1,所以gcd(a’,a’+b’)=1，所以一定满足$a'+b'|d$。同时因为d(a’+b’)<=n,所以a’+b’ <=N−−√ <script type="math/tex" id="MathJax-Element-492"><=\sqrt{N}</script>，同时因为d=x(a’+b’),d(a’+b’)<=n,所以$x(a'+b')^2<=n$,那么满足$1<=x<={N\over{(a'+b')^2}}$，即d有${N\over{(a'+b')^2}}$ 种取值。所以我们枚举a’+b’=k，那么显然要计算gcd(a’,b’)=1的对数,等价于求gcd(a’,k)=1的对数，所以就是$\phi{（k）}$。用O($\sqrt{N}$)欲处理一下即可。最后答案$ans=\sum^{\sqrt{N}}_{k=1}{\phi{(k)}*{N\over{k^2}}}$

Code

#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#define ll long long
using namespace std;
const ll maxn=1e3+5; 
ll n,i,t,j,k,l,ans,d[maxn],f[maxn];
bool bz[maxn];
int main(){
    freopen("uria.in","r",stdin);freopen("uria.out","w",stdout);
    scanf("%lld",&n);
    t=sqrt(n);
    for (i=2;i<=t;i++){
        if (!bz[i]) d[++d[0]]=i,f[i]=i-1;
        for (j=1;j<=d[0];j++){
            if (i*d[j]>t) break;
            bz[i*d[j]]=true;
            if (!(i%d[j])){
                f[i*d[j]]=f[i]*d[j];break;
            }
            f[i*d[j]]=f[i]*(d[j]-1);
        }
    }
    for (i=1;i<=t;i++)
        k=n/(i*i),ans+=k*f[i];
    printf("%lld\n",ans);
}