ARC116 C - Multiple Sequences（质因子分解+组合数学）

题意：

解法：

考虑枚举a[n],当固定a[n]之后,设a[n]=x,
假设x的质因子为(p1,c1),(p2,c2)...(pm,cm),其中p是质因子,c是该质因子的个数.

那么问题相当于将质因子分配到[1,n]的n个盒子里,盒子可以为空,
n个盒子从左到右依次乘过来就是a[n]=x了.

但是质因子有的相同有的不同,我们对每种质因子(pi,ci)单独考虑:
对于一种质因子(pi,ci),相当于ci个相同的球丢入n个不同的盒子,方案数为C(n+ci-1,ci),
我们将每种质因子的方案数相乘就是当前a[n]=x的答案了.

从[1,m]枚举a[n],每次计算答案即可,复杂度O(m*sq).

code：

#include<bits/stdc++.h>
#define int long long
using namespace std;
const int maxm=2e6+5;
const int mod=998244353;
int fac[maxm],inv[maxm];
int n,m;
int ppow(int a,int b,int mod){
    int ans=1%mod;a%=mod;
    while(b){
        if(b&1)ans=ans*a%mod;
        a=a*a%mod;
        b>>=1;
    }
    return ans;
}
void init(){
    fac[0]=1;
    for(int i=1;i<maxm;i++)fac[i]=fac[i-1]*i%mod;
    inv[maxm-1]=ppow(fac[maxm-1],mod-2,mod);
    for(int i=maxm-2;i>=0;i--)inv[i]=(i+1)*inv[i+1]%mod;
}
int C(int n,int m){
    if(m<0||m>n)return 0;
    return fac[n]*inv[m]%mod*inv[n-m]%mod;
}
void solve(){
    init();
    cin>>n>>m;
    int ans=0;
    for(int i=1;i<=m;i++){//枚举a[n]
        int x=i;
        int sum=0;
        vector<int>temp;
        for(int j=2;j*j<=x;j++){//质因子分解
            if(x%j==0){
                int cnt=0;
                while(x%j==0){
                    x/=j;
                    cnt++;
                }
                sum+=cnt;
                temp.push_back(cnt);
            }
        }
        if(x!=1){
            sum++;
            temp.push_back(1);
        }
        int t=1;
        for(auto p:temp){
            t=t*C(n+p-1,p)%mod;
        }
        ans=(ans+t)%mod;
    }
    cout<<ans<<endl;
}
signed main(){
    ios::sync_with_stdio(0);
    solve();
    return 0;
}

---