第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛 F-等式

第13届景驰-埃森哲杯广东工业大学ACM程序设计大赛 F-等式

链接：https://www.nowcoder.com/acm/contest/90/F

来源：牛客网

题目描述

给定n，求1/x + 1/y = 1/n （x<=y）的解数。（x、y、n均为正整数）

输入描述:

在第一行输入一个正整数T。

接下来有T行，每行输入一个正整数n，请求出符合该方程要求的解数。

（1<=n<=1e9）

输出描述:

输出符合该方程要求的解数。

示例1

输入

3
1
20180101
1000000000

输出

1
5
181

---

思路

$$\begin{eqnarray} \cfrac{1}{x} + \cfrac{1}{y} &=&\cfrac{1}{n} \\ ny+nx&=&xy \\ (n-x)(n-y)&=&n^2 \end{eqnarray}$$

转化为求 $n^2$ 的因子 小于等于 n 的个数。

将 n 分解质因子，复杂度为 logn，$n^2$ 对应的质因子和 n 相同，且个数翻倍，即可得到 $n^2$ 的质因子个数。

根据数论中的唯一分解定理可得：

$$\begin{eqnarray} &任何&整数n都可以表示为：\\ &n=&p_1^{e_1}*p_2^{e_2}*\ldots*p_n^{e_n},其中p_1,p_2,\ldots,p_n都为素数 \\ &并且&n的约数个数为(1+e_1)*(1+e_2)*\ldots*(1+e_n) \\ &\therefore& n^2 =(p_1^{e_1}*p_2^{e_2}*\ldots*p_n^{e_n})^2 = (p_1^{2e_1})*(p_2^{2e_2})*\ldots*(p_n^{2e_n}) \\ &\therefore& 因子个数为(1+2e_1)*(1+2e_2)*\ldots*(1+2e_n) \\ &\therefore& 可以利用唯一分解定理求出e_1,e_2,\ldots,e_n \end{eqnarray}$$

设 $n^2$ 的因子个数为 ans, $n^2$ 的所有因子中除n外都是成对出现的，故方程解数为 (ans+1)/2。

AC代码

#include <iostream>
#include<cstring>

using namespace std;

int num[64];

int main() {
    int T;
    cin >> T;
    while(T--) {
        int n;
        cin >> n;
        memset(num, 0, sizeof(num));
        int p=0;
        for (int i=2; i*i<=n; i++) {
            if(n%i==0) {
                while (n%i == 0) {
                    num[p]++;
                    n /= i;
                }
                p++;
            }
        }

        if(n!=1) {
            num[p]++;
            p++;
        }
        for(int i=0; i<p; i++) {
            num[i]*=2;
        }
        int ans=1;
        for (int i=0; i<p; i++) {
            ans*=(num[i]+1);
        }
        cout << (ans+1)/2 << endl;
    }
    return 0;
}