題目大意

給出一個長度為 \(n\) 的序列 \(a\)，令 \(m=\prod_{i=1}^na_i\)，問有多少個長度為 \(n\) 的序列使得序列中的所有數的乘積等於 \(m\)。

輸出方案數模 \(10^9+7\)。

解題思路

前置知識：組合數學，逆元。

先考慮將 \(a_i\) 分解質因數，並記錄每個質數出現的質數，用 map 維護即可。

假設現在有 \(n\) 個箱子。

對於每個質數， 都可以分別放到 \(n\) 個箱子上去， 可以在某個箱子不放， 也可以在某個箱子放很多個。

根據隔板法，得：若一種質因數共有 \(x\) 個， 那麼就有 \(\operatorname{C}(n+x-1,n-1)\)

設有 \(p\) 個不同的質數，第 \(i\) 個質數出現了 \(q_i\) 次，那麼，

CODE

#include <bits/stdc++.h>

using namespace std;

#define int long long

const int _ = 4e5;

bool pri[_ + 7];

map<int, int> num;

int fac[_ + 7], invfac[_ + 7];

const int Mod = 1e9 + 7;

inline void getprime()
{
    for (int i = 2; i <= _; i++)
        if (!pri[i])
            for (int j = i + i; j <= _; j += i)
                pri[j] = 1;
}

inline void getnum(int x)
{
    for (int i = 2; i * i <= x; i++)
        if (!pri[i])
            while (x % i == 0)
            {
                x /= i;
                num[i]++;
            }
    if (x != 1)
        num[x]++;
}

inline int Pow(int x, int y)
{
    int res = 1;
    while(y)
    {
    	if(y & 1) res = res * x % Mod;
    	x = x * x % Mod;
    	y >>= 1;
	}
	return res;
}

inline void getfac()
{
    fac[0] = 1;
    for (int i = 1; i <= _; i++)
        fac[i] = fac[i - 1] * i % Mod;
    invfac[_] = Pow(fac[_], Mod - 2);
    for (int i = _ - 1; i >= 0; i--)
        invfac[i] = invfac[i + 1] * (i + 1) % Mod;
}

inline long long C(int a, int b)
{
    if (b == 0)
        return 1;
    return fac[a] * invfac[b] % Mod * invfac[a - b] % Mod;
}

int n, tmp;

int ans = 1;

signed main()
{
    getprime();
    getfac();
    scanf("%lld", &n);
    for (int i = 1; i <= n; i++)
    {
        scanf("%lld", &tmp);
        getnum(tmp);
    }
    for (auto it = num.begin(); it != num.end(); it++)
    {
        ans *= C(n + (it->second) - 1, n - 1);
        ans %= Mod;
    }
    printf("%lld\n", ans % Mod);
    return 0;
}
 

本文來自部落格園，作者：蒟蒻orz，轉載請註明原文連結：https://www.cnblogs.com/orzz/p/15585403.html