Newcoder 141 H.Diff-prime Pairs（数论）

Description

给出一整数$n$，找二元组$(i,j)$的对数，使其满足$\frac{i}{gcd(i,j)},\frac{j}{gcd(i,j)}$均为素数且$1\le i,j\le n$

Input

一个整数$n(1\le n\le 10^7)$

Output

输出满足条件的二元组个数

Sample Input

3

Sample Output

2

Solution

答案即为寻找满足$(d\cdot p,d\cdot q)(d\cdot p,d\cdot q\le n)$的对数，其中$p,q$为互不相同的素数，考虑求$p<q$的对数，乘二即为答案，显然$d\le \lfloor\frac{n}{q}\rfloor$，假设$n$以内所有的素数为$p_1,...,p_m$，那么$p_i$作为较大素数对答案的贡献为$(i-1)\cdot \lfloor\frac{n}{p_i}\rfloor$，故答案为$2\cdot\sum\limits_{i=2}^m(i-1)\cdot \lfloor\frac{n}{p_i}\rfloor$，$O(n)$线性筛出 n n <script type="math/tex" id="MathJax-Element-18">n</script>以内所有素数即可

Code

#include<cstdio>
#include<vector>
using namespace std;
typedef long long ll;
#define maxn 10000005
vector<int>p;
int mark[maxn]; 
void get_prime(int n)
{
    for(int i=2;i<=n;i++)
    {
        if(!mark[i]) 
        {
            mark[i]=i;
            p.push_back(i);
        } 
        for(int j=0;j<p.size()&&p[j]*i<=n;j++)
        {
            mark[i*p[j]]=p[j];
            if(i%p[j]==0) 
                break;
        }
    }
} 
int main()
{
    int n;
    scanf("%d",&n);
    get_prime(n);
    ll ans=0;
    for(int i=1;i<p.size();i++)ans+=(ll)i*(n/p[i]);
    printf("%lld\n",2ll*ans);
    return 0;
}