ARC116C Multiple Sequences(隔板法，枚举)

ARC116C(隔板法，枚举)

题目链接

Problem Statement

Given are integers $N$ and $M$. How many sequences $A$ of $N$ integers satisfy the following conditions?

• $1\le A_i\le M\left(i=1,2,\dots ,N\right)$
• $A_{i+1}$ is a multiple of $A_i$. $\left(i=1,2,\dots ,N-1\right)$

Since the answer can be enormous, report it modulo $998244353$.

Constraints

• All values in input are integers.
• $1\le N\le 2\times 10^5$
• $1\le M\le 2\times 10^5$

题意

给你$N$和$M$,问你有多少个长度为$N$的$A$序列满足下述条件：

• $1\le A_i\le M(i=1,2,\dots ,N)$
• $A_{i+1}\%A_i==0(i=1,2,\dots ,N)$

思路

​ 我们很容易想到这个序列一定是与这个序列的最后一个数字$a_N$有关的，因此我们去枚举最后一位数从$1到M$,同时也是跟其$a_n=p_1^{q_1}{p}_2^{q_2}{p}_3^{q_3}\dots p_n^{q_n}(p_1<p_2<p_3<\cdots <p_n)$$(p为素数，q为a_n中含有p_i的最高次方)$那么我们可以想到$a_i$与$a_{i+1}$的关系就是$a_i$乘上$a_n$中含有的某些素数次方所得到的，或者什么也不取，最后只有凑到$a_n$就行了。那么想到隔板法，一个素数一共最多有$q_1$个可以取，你每一次取得就像隔板，隔板左边的就是你已经取的个数，隔板右边的就是你还没有取的个数，那么这个问题就相当于$N$个不同的盒子，$q_i$个小球，问你一共有几种方案，就是说每种素数的取法就为$C_{n+q_i+1}^{n-1}$,就是将枚举最后一位的素数方案数相乘，然后枚举加起来就行了

代码

#include<bits/stdc++.h>
#define Case int t;scanf("%d",&t);while(t--)
using namespace std;
typedef long long ll;
const int N=1e6+10;
const int M=1e6+10;
const ll mod=998244353;
ll a[N],b[N];
ll fac[N], inv[N];
ll qpow(ll q,ll n){
   ll res=1;
   while(n){
      if(n&1)res=(res*q)%mod;
      n>>=1;
      q=(q*q)%mod;
   }
   return res;
}
void init(){
   fac[0]=1;
   for(int i=1;i<N;i++){
      fac[i]=fac[i-1]*i%mod;
   }
   inv[N-1]=qpow(fac[N-1],mod-2);
   for(int i=N-2;i>=0;i--){
      inv[i]=inv[i+1]*(i+1)%mod;
   }
}
void run(){
   ll n,m;
   scanf("%lld%lld",&n,&m);
   ll ans=0;
   for(int i=1;i<=m;i++){
      ll e=i;
      vector<int>tmp;
      for(int j=2;j*j<=e;j++){
         if(e%j==0){
            int cnt=0;
            while(e%j==0){
               cnt++;
               e/=j;
            }  
            tmp.push_back(cnt);
         }
      }
      if(e>1){
         e=1;
         tmp.push_back(1);
      }
      ll temp=1;
      for(auto it:tmp){
         temp=(temp*fac[n+it-1]%mod*inv[n-1]%mod*inv[it]%mod)%mod;
      }
      ans=(ans+temp)%mod;
   }
   printf("%lld\n",ans);
}
int main(){
   init();
   // Case
   run();
   return 0;
}

道阻且长，且行且珍惜