传送门:1499D
标签:数学
题目大意
给你三个正整数(大于零) c , d c, d c,d 和 x x x。你需要找出满足等式 c ⋅ l c m ( a , b ) − d ⋅ g c d ( a , b ) = x c·lcm(a,b)-d·gcd(a,b)=x c⋅lcm(a,b)−d⋅gcd(a,b)=x 的正整数对 ( a , b ) (a, b) (a,b) 的数量。其中 l c m ( a , b ) lcm(a, b) lcm(a,b) 是 a a a 和 b b b 的最小公倍数, g c d ( a , b ) gcd(a, b) gcd(a,b) 是 a a a 和 b b b 的最大公约数。
输入:第一行包含一个整数 t t t( 1 ≤ t ≤ 1 0 4 1≤t≤10^4 1≤t≤104)—— 测试用例的数量。每个测试用例都包含一行,包含三个整数 c , d c, d c,d 和 x x x( 1 ≤ c , d , x ≤ 1 0 7 1≤c,d,x≤10^7 1≤c,d,x≤107)。
输出:对于每个测试用例,打印一个整数 —— 满足上述等式的 ( a , b ) (a, b) (a,b) 对的数量。
算法分析
- 如果将 a a a 表示为 A g Ag Ag,将 b b b 表示为 B g Bg Bg,则其中 g = g c d ( a , b ) g=gcd(a,b) g=gcd(a,b) 且 g c d ( A , B ) = 1 gcd(A,B)=1 gcd(A,B)=1。根据定义 l c m ( a , b ) = a b g lcm(a,b)=\frac{ab}{g} lcm(a,b)=gab,所以我们可以将 l c m ( a , b ) lcm(a,b) lcm(a,b) 表示为 A g B g = A B g \frac{AgB}{g}=ABg gAgB=ABg。然后可以重写陈述中的方程如下: c ⋅ A B g − d ⋅ g = x ⟹ g ( c ⋅ A B − d ) = x c·ABg-d·g=x \implies g(c·AB-d)=x c⋅ABg−d⋅g=x⟹g(c⋅AB−d)=x。由于左侧可被 g g g 整除,右侧也应如此。所以我们可以遍历 g g g 作为 x x x 的约数。如果右侧的 c ⋅ A B = x g + d c·AB=\frac{x}{g}+d c⋅AB=gx+d 不可被 c c c 整除,则可以跳过这样的 g g g。 A B = x c + d c AB=\frac{x}{c}+\frac{d}{c} AB=cx+cd(我们称它为 k k k)。如果 k k k 具有素数因子 p p p,则 A A A 或 B B B 应该可被 p p p 整除,因为 g c d ( A , B ) = 1 gcd(A,B)=1 gcd(A,B)=1( A A A 和 B B B 没有公共因子)。因此对于当前的 g g g 值,有 2 z 2^z 2z 对 ( A (A (A 和 B ) B) B),其中x为k的素数因子个数。
- 我们可以预计算每个数字的最大素因数(可能需要的最大值为 k k k)上的最小素因数,在 O ( 2 ⋅ 1 0 7 l o g l o g ( 2 ⋅ 1 0 7 ) ) O(2·10^7loglog(2·10^7)) O(2⋅107loglog(2⋅107)) 时间内使用欧拉筛法。那么在 O ( x l o g x ) O(\sqrt{x} log x) O(xlogx) 时间内可以为每个测试用例解决这个问题,但这还不够快。为了加快这个方法,我们可以预计算每个数字的最大素因数的数量。令 m i n d i mind_i mindi 为 i i i 的最小素因数, v a l i val_i vali 为 i i i 的素因数数量。那么 v a l i = v a l i / m i n d i val_i=val_i/mind_i vali=vali/mindi 加上 1,如果 m i n d i ≠ m i n d i / m i n d i mind_i≠mind_i/mind_i mindi=mindi/mindi。现在要解决问题,我们只需遍历 x x x 的约数,所以每个测试用例的时间复杂度为 O ( x ) O(\sqrt{x}) O(x)。
代码实现
#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 1e7+5;
int tt;
int c,d,x;
int pr[2*N],iki[22];
ll ans;
void f(int nw){
if((x/nw+d) % c) return;
int k = (x/nw+d)/c;
ans += iki[pr[k]];
}
int main(){
ios::sync_with_stdio(false);
cin.tie(0);cout.tie(0);
iki[0] = 1;
for(int i = 1;i <= 20;i++) iki[i] = iki[i-1] * 2;
for(int i = 2;i < 2*N;i++){
if(pr[i]) continue;
for(int j = i;j < 2*N;j += i) pr[j]++;
}
cin >> tt;
while(tt--){
cin >> c >> d >> x;
ans = 0;
for(int i = 1;i*i <= x;i++){
if(x % i) continue;
f(i);
if(i != x/i) f(x/i);
}
cout<<ans<<'\n';
}
}