Diff-prime Pairs 牛客

牛客 Diff-prime Pairs

题意:给定$N(1\le N\le 10^7)$，求有多少对数$(i,j)$满足$\frac{i}{gcd(i,j)}$和$\frac{j}{gcd(i,j)}$均为素数，$(i_1,j_1),(i_2,j_2)$不同当且$i_1!=i_{2}or{j}_1!=j_2$

Sol:

$\frac{i}{g}=prime$ $\frac{j}{g}=prime$

$i=prime\ast g$ $j=prime\ast g$

$i\in [1,n]$ $j\in [1,n]$

这样的$i,j$要么是素数对，要么是素数对的倍数

枚举$g$，则$\frac{n}{g}$表示$prime$的范围，如果在$\frac{n}{g}$范围内有$k$个素数

则说明有$|(g{p}_1,g{p}_2,...,g{p}_k)\times (g{p}_1,g{p}_2,...,g{p}_k)|-k$对组合是符合要求的

且$i,j\in g{p}_1,g{p}_2,...,g{p}_k$

所以枚举$g$可以表示出所有素数对和他们的倍数

#include <bits/stdc++.h>
using namespace std;
const int N=1e7+10;
int prime[N],st[N],sum[N];
int cnt;
void get_primes(int n)
{
     for(int i=2;i<=n;i++)
     {
     	if(!st[i]) prime[++cnt]=i,sum[i]=1;

            for(int j=1;i*prime[j]<=n;j++)
            {
                st[i*prime[j]]=1;
                if(i%prime[j]==0) break;
            }
        
     }
}
int main()
{     
	 ios::sync_with_stdio(0);
	 cin.tie(0);
	 int n;
	 cin>>n;
	 get_primes(n);
  
    for(int i=1;i<=n;i++) sum[i]=sum[i-1]+sum[i];
	 long long ans=0;
     ans=1LL*(cnt-1)*cnt;
	 cout<<ans<<endl;
	 return 0;
}