5648. 生成乘积数组的方案数(质因数分解&组合数学)

最新推荐文章于 2023-01-02 14:45:10 发布

酷酷的Herio

最新推荐文章于 2023-01-02 14:45:10 发布

阅读量424

点赞数 1

分类专栏：组合数学数论

本文链接：https://blog.csdn.net/weixin_45750972/article/details/113083686

版权

数论同时被 2 个专栏收录

105 篇文章 2 订阅

订阅专栏

组合数学

41 篇文章 1 订阅

订阅专栏

5648. 生成乘积数组的方案数(质因数分解&组合数学)

思路：质因数分解+组合数学。

将 $k$ 进行质因数分解后，考虑对于某一个质因数的个数 $c n t$ ，如何分配给个 $n$ 个位置，可以为空。

这等价于把 $c n t$ 个球放入 $n$ 个篮子，篮子可以为空的方案。

我们可以先将 $n$ 个篮子都加入一个球，问题等价于 $c n t + n$ 个球放入 $n$ 个篮子，篮子不能为空的方案数，这可以用隔板法解决。

$a n s = C (c n t + n - 1, n - 1) = C (c n t + n - 1, c n t)$ ，在 $c n t + n - 1$ 个可以放隔板的位置放入 $n - 1$ 个隔板，这样就分成了 $n$ 个部分。

最后答案都乘起来就行了。

代码

typedef long long ll;
const int N=1e4+5,mod=1e9+7;
ll fac[N];
class Solution {
public:
    ll ksm(ll a,ll n){
        ll s=1;
        while(n){
            if(n&1) s=s*a%mod;
            a=a*a%mod;
            n>>=1;
        }return s;
    }
    ll C(ll m,ll n){
        return fac[m]*ksm(fac[n]*fac[m-n]%mod,mod-2)%mod;
    }
    vector<int> waysToFillArray(vector<vector<int>>& q) {
        fac[0]=1;for(int i=1;i<N;i++) fac[i]=fac[i-1]*i%mod;
        vector<int>res;
        for(auto v:q){
            ll ans=1;
            int n=v[0],k=v[1];
            for(int i=2;i*i<=k;i++){
                if(k%i==0){
                    int s=0;
                    while(k%i==0) k/=i,s++;
                    ans=ans*C(n+s-1,s)%mod;
                }
            }
            if(k>1) ans=ans*n%mod;
            res.push_back(ans);
        }  
        return res;
    }   
};