【ACWing】871. 约数之和

题目地址：

https://www.acwing.com/problem/content/873/

给定$n$个正整数$a_i$，输出它们的乘积的所有约数之和。答案对$10^9+7$取模。

输入格式：
第一行包含整数$n$。接下来$n$行，每行包含一个整数$a_i$。

输出格式：
输出一个整数，表示所给正整数的乘积的约数之和，答案需对$10^9+7$取模。

数据范围：
$1\le n\le 100$
$1\le a_i\le 2\times 10^9$

对于一个正整数$x$，若其质因子分解是$x=p_1^{{\alpha }_1}{p}_2^{{\alpha }_2}...p_k^{{\alpha }_k}$，那么$x$的约数个数是$(1+p_1+p_1^2+...+p_1^{{\alpha }_1})(1+p_2+p_2^2+...+p_2^{{\alpha }_2})...(1+p_k+p_k^2+...+p_k^{{\alpha }_k})$。我们可以统计每个$a_i$的各个质因子个数，最后按照上面公式求乘积的约数个数。代码如下：

#include <iostream>
#include <unordered_map>
using namespace std;

const int MOD = 1e9 + 7;
unordered_map<int, int> map;

// 套用分解质因数模板
void divide(int n) {
    for (int i = 2; i <= n / i; i++) {
        if (n % i == 0) {
            int c = 0;
            while (n % i == 0) {
                c++;
                n /= i;
            }
            
            map[i] += c;
        }
    }

    if (n >= 2) map[n]++;
}

int main() {
    int n;
    cin >> n;
    while (n--) {
        int x;
        cin >> x;

        divide(x);
    }

    long res = 1;
    for (auto pa : map) {
        int p = pa.first, a = pa.second;
        long t = 1;
        while (a--) t = (t * p + 1) % MOD;
        res = res * t % MOD;
    }

    cout << res << endl;

    return 0;
}

时间复杂度$O(\sum {\sqrt{a}}_i)$，空间$O(k)$，$k$为不同质因子总个数。