[Acwing] 200. Hankson的趣味题 预处理+质因数分解

前言

传送门 :

思路

给定$a_0,a_1,b_0,b_1$求有多少个$x$满足

$gcd(x,a_0)=a_1$
$lcm(x,b_0)=b_1$

显然的,我们可以从$lcm(x,b_0)=b_1,x是b_1的约数$

因此我们可以暴力 所有$b_1$的约数,然后对于每个约数都$check$

时间复杂度$O(T\ast logn\ast \sqrt{n})$ 大约是$8\ast 10^8>10^8$ 因此是会被卡的

因此我们考虑优化

显然我们能优化的只是$\sqrt{n}$

我们可以通过预处理$1-N$的所有质数,根据质数定理只有$\frac{\sqrt{N}}{ln\sqrt{N}}$ 个

然后通过枚举所有的质数,对$b1$进行质因数分解,最后通过$dfs$将所有约数找出来

因此时间复杂度降到了$O(T\ast logn\ast \frac{\sqrt{N}}{ln\sqrt{N}})$大约$4\ast 10^7$

Mycode

const int N = 4e4+5000, M = 50;
int primes[N] ,cnt;
bool st[N];

pii factor[N];
int cntf;


int divider[N],cntd;


//预处理出所有的质数
void get_primes(int n){
	for(int i=2;i<=n;i++){
		if(!st[i]) primes[cnt++] = i ;
		for(int j=0;primes[j]<=n/i;j++){
			st[primes[j]*i] =true;
			if(i%primes[j] == 0) break;
		}
	}
}

/**
这里是将已经处理好的 质因数
dfs求出所有的约数 
**/
void dfs(int u,int p){
	if(u>cntf){
		divider[cntd++] =  p;
		return;
	}
	
	for(int i = 0;i<= factor[u].second;i++){
		dfs(u+1,p);
		p*=factor[u].x;//
	}
}

void solve(){
	int a0,a1,b0,b1;
	cin>>a0>>a1>>b0>>b1;
	
	int d =  b1;//用于分解质因数
	cntf = 0 ;
	//b1 的所有约数
	for(int i=0;primes[i]<=d/primes[i];i++){
		int p  = primes[i];
		if(d%p == 0){
			int s = 0 ;
			while(d%p == 0) s++,d/=p;
			factor[++cntf] = {p,s};//分解出来所有约数
		} 
	}
	if(d>1) factor[++cntf] = {d,1};
	
	cntd = 0 ;
	dfs(1,1);
	
	
	int res = 0 ;
	for(int i=0;i<cntd;i++){
		int x = divider[i];
		if(__gcd(x,a0) == a1 && 1ll*x*b0/__gcd(x,b0) == b1)++res;
	}
	
	cout<<res<<endl;
	
	
}

int main(){
	get_primes(N);
	
    int t;cin>>t;while(t--)
    solve();
    return 0 ;
}