ARC116 C - Multiple Sequences【组合数学/DP】

Link

题目

给出一个 $n$ 和 $m$，求出满足下列要求的序列的个数：

• $1\le A_i\le M(i=1,2,\cdots ,N)$
• $A_{i+1}$ 是 $A_i$ 的倍数$(i=1,2,\cdots ,N-1)$

结果取模 $998244353$，$1\le N,M\le 2\times 10^5$。

分析

组合数学解法

考虑枚举 $A_n$ 的值，然后把所有取值的方案加起来。假设 $A_n=x$，为了满足性质 $2$，对 $x$ 进行质因子分解，得到每个质因子 $p_i$ 及其幂 $e_i$。对每个质因子分别考虑，然后把每个质因子的结果乘起来。对于 $e_i$ 个质因子 $p_i$，需要把它放置在 $n$ 个位置中（放置在第 $i$ 个位置表示 $A_i$ 的值在 $A_{i-1}$ 的基础上乘上 $p_i^k$，$k$ 表示可以放多个）。另外注意是 n 个位置不是 $n-1$，因为质因子的个数不一定要用完。自此，问题转化为将 $e$ 个相同的物体放入 $n$ 个不同的盒子且盒子可以空的问题模型，答案是 $\left(\begin{array}{c}n+e-1\\ e\end{array}\right)$。最后的复杂度为 $O(m{\log }_{2}m)$。预处理阶乘和阶乘逆元计算组合数时，处理的范围要大于 $N$。

代码

#include <vector>
#include <algorithm>
#include <cstdio>
using namespace std;
typedef long long ll;
const int mod=998244353;
const int N=2e5+200;
ll fac[N],inv[N];
ll power(ll x,int y){
    ll res=1;
    while(y){
        if(y&1) res=res*x%mod;
        x=x*x%mod;
        y>>=1;
    }
    return res;
}
void init(){
    int maxn=2e5+100;//要比2e5大
    fac[0]=1;
    for(int i=1;i<=maxn;i++) fac[i]=fac[i-1]*i%mod;
    inv[maxn]=power(fac[maxn],mod-2);
    for(int i=maxn-1;i>=0;i--) inv[i]=inv[i+1]*(i+1)%mod;
}
ll solve(int a,int b){
    if(b>a||b<0) return 0;
    ll res=fac[a]*inv[b]%mod*inv[a-b]%mod;
    return res;
}
int main(){
    int n,m;
    scanf("%d%d",&n,&m);
    ll ans=0;
    init();
    for(int i=1;i<=m;i++){//枚举A[n]
        int x=i;
        ll res=1;
        vector<int>p;
        for(int j=2;1LL*j*j<=x;j++){
            if(x%j==0){
                int cnt=0;
                while(x%j==0){
                    x/=j;
                    cnt++;
                }
                p.push_back(cnt);
            }
        }
        if(x>1) p.push_back(1);
        for(auto e:p)
            res=res*solve(n+e-1,e)%mod;
        ans=(ans+res)%mod;
    }
    printf("%lld\n",ans);
    return 0;
}