Educational Codeforces Round 106 (Rated for Div. 2) D. The Number of Pairs (gcd,计数问题)

题目链接：https://codeforces.com/problemset/problem/1499/D
题意：给三个正整数$c,d,x$，求有多少对$(a,b)$满足$c\ast lcm(a,b)-d\ast gcd(a,b)=x$。
思路：我们可以看到左边的式子有一个因子为$gcd(a,b)$，所以x应该为$gcd(a,b)$的倍数，我们把式子两边同时除$gcd(a,b)$得到
$$c\ast (lcm(a,b)/gcd(a,b))-d=x/gcd(a,b)$$
对于$gcd(a,b)$我们可以枚举x的因子$t$作为$gcd(a,b)$，并且设$lcm(a,b)/gcd(a,b)=k$，得到：
$$c\ast k-d=x/t$$
此时：问题就转化为：已知$gcd(a,b)=t,lcm(a,b)/gcd(a,b)=k$求有多少对(a,b)满足条件，我们发现：k应该是两个互质的数的乘积，此时问题可以转化为：$k$可以写为几对互质的数的乘积？这个问题就很简单了，对于一个质因子个数为p的数k它可以写为$2^p$对互质的数的乘积（每个质因子都可能放在左边或者右边）
这时问题就解决了.计算复杂度:对每个数的质因子个数可以O(N)预处理.枚举因子$O(sqrt(x))$,每次计算2的幂次是$O(log(p))$,总复杂度为$O(N+t\ast sqrt(x)\ast log(m))$ $m=max(p)$
代码:

#include<bits/stdc++.h>
#define ll long long
using namespace std;
const int N=2e7+7;
ll c,d,x;
ll p[N],st[N],f[N],cnt;
void init(int n)
{
    for (int i = 2; i <= n; i ++ )
    {
        if (!st[i])
        {
            p[cnt ++ ] = i;
            f[i] = 1;
        }
        for (int j = 0; p[j] *i<= n; j ++ )
        {
            int t = p[j] * i;
            st[t] = true;
            if (i % p[j] == 0)
            {
                f[t] = f[i];
                break;
            }
            f[t] = f[i] + 1;
        }
    }
}
ll cal(ll x)
{
	if((x+d)%c) return 0;
	else 
	{
//		printf("x:%lld  %lld\n",x,(x+d)/c);
		return 1ll<<f[(x+d)/c];
	}
}
int main()
{
	ll t;
	scanf("%lld",&t);
	init(N-7);
//	for(int i=1;i<=20;i++) printf("%lld ",f[i]);
//	printf("\n");
	while(t--)
	{
		scanf("%lld%lld%lld",&c,&d,&x);
		ll ans=0;
		for(ll i=1;i*i<=x;i++)
		{
			if(x%i==0)
			{
				ans+=cal(i);
//				printf("ans:%lld\n",ans);
				if(i*i!=x) ans+=cal(x/i);
//				printf("ans:%lld\n",ans);
			}
		}
		printf("%lld\n",ans);
	}
}