题意:有一个整数 x ∈ [ 0 , n ] x\in[0,n] x∈[0,n],取 i i i的概率为 p i p_i pi。执行 m m m次操作,每次把 x x x等概率变成 [ 0 , x ] [0,x] [0,x]中的一个整数,求操作完后等于每个数的概率。模 998244353 998244353 998244353。
n ≤ 1 0 5 , m ≤ 1 0 18 n\leq 10^5,m\leq10^{18} n≤105,m≤1018
显然有一个dp式子
f i , j = ∑ k = j n f i − 1 , k k + 1 f_{i,j}=\sum_{k=j}^n\frac{f_{i-1,k}}{k+1} fi,j=k=j∑nk+1fi−1,k
考虑生成函数
f i ( x ) = ∑ j = 0 n ∑ k = j n [ x k ] f i − 1 ( x ) k + 1 x j f_i(x)=\sum_{j=0}^n\sum_{k=j}^n\frac{[x^k]f_{i-1}(x)}{k+1}x^j fi(x)=j=0∑nk=j∑nk+1[xk]fi−1(x)xj
= ∑ k = 0 n [ x k ] f i − 1 ( x ) k + 1 ∑ j = 0 k x j =\sum_{k=0}^n\frac{[x^k]f_{i-1}(x)}{k+1}\sum_{j=0}^kx^j =k=0∑nk+1[xk]fi−1(x)j=0∑kxj
等比数列求和
= ∑ k = 0 n [ x k ] f i − 1 ( x ) k + 1 x k + 1 − 1 x − 1 =\sum_{k=0}^n\frac{[x^k]f_{i-1}(x)}{k+1}\frac{x^{k+1}-1}{x-1} =k=0∑nk+1[xk]fi−1(x)x−1xk+1−1
似乎推不动了
但是有奥妙重重的一步:
= ∑ k = 0 n [ x k ] f i − 1 ( x ) x − 1 x k + 1 − 1 k + 1 =\sum_{k=0}^n\frac{[x^k]f_{i-1}(x)}{x-1}\frac{x^{k+1}-1}{k+1} =k=0∑nx−1[xk]fi−1(x)k+1xk+1−1
看起来没啥用,但这样把右边写成了积分的形式
= ∑ k = 0 n [ x k ] f i − 1 ( x ) x − 1 ∫ 1 x t k d t =\sum_{k=0}^n\frac{[x^k]f_{i-1}(x)}{x-1}\int_1^xt^kdt =k=0∑nx−1[xk]fi−1(x)∫1xtkdt
然后盯着这个式子:
∑ k = 0 n [ x k ] f i − 1 ( x ) ∫ 1 x t k d t \sum_{k=0}^n[x^k]f_{i-1}(x)\int_1^xt^kdt k=0∑n[xk]fi−1(x)∫1xtkdt
t t t可能看不习惯,换个更通用的写法:
∑ k = 0 n [ x k ] f ( x ) ∫ L R x k d x \sum_{k=0}^n[x^k]f(x) \int_L^Rx^kdx k=0∑n[xk]f(x)∫LRxkdx
相当于把多项式拆成单项式,每个单项式对区间 [ L , R ] [L,R] [L,R]求定积分,最后还贴心地帮你乘上了系数
那不就是对多项式求积分吗
∫ L R f ( x ) d x \int_L^Rf(x)dx ∫LRf(x)dx
代回原式:
f i ( x ) = 1 x − 1 ∫ 1 x f i − 1 ( t ) d t f_i(x)=\frac{1}{x-1}\int_1^xf_{i-1}(t)dt fi(x)=x−11∫1xfi−1(t)dt
然而这个积分下界是 1 1 1,不好处理
然后又是奥妙重重的一步:
令 g i ( x ) = f i ( x + 1 ) g_i(x)=f_i(x+1) gi(x)=fi(x+1)
g i ( x ) = 1 x ∫ 1 x + 1 f i − 1 ( t ) d t g_i(x)=\frac{1}{x}\int_1^{x+1}f_{i-1}(t)dt gi(x)=x1∫1x+1fi−1(t)dt
= 1 x ∫ 0 x f i − 1 ( t + 1 ) d t =\frac{1}{x}\int_0^{x}f_{i-1}(t+1)dt =x1∫0xfi−1(t+1)dt
= 1 x ∫ 0 x g i − 1 ( t ) d t =\frac{1}{x}\int_0^{x}g_{i-1}(t)dt =x1∫0xgi−1(t)dt
那就 g i − 1 g_{i-1} gi−1的不定积分在 x x x处的取值,容易得出
g i , j = 1 j + 1 g i − 1 , j g_{i,j}=\frac{1}{j+1}g_{i-1,j} gi,j=j+11gi−1,j
这样如果知道 g 0 g_0 g0就可以轻松算出 g m g_m gm
而 f f f和 g g g直接二项式定理暴拆再卷积一下就可以互相转换,问题解决
复杂度 O ( n log n ) O(n\log n) O(nlogn)
#include <iostream>
#include <cstdio>
#include <cstring>
#include <cctype>
#define MAXN 262144
using namespace std;
const int MOD=998244353;
typedef long long ll;
inline int read()
{
int ans=0;
char c=getchar();
while (!isdigit(c)) c=getchar();
while (isdigit(c)) ans=(ans<<3)+(ans<<1)+(c^48),c=getchar();
return ans;
}
inline int qpow(int a,int p)
{
int ans=1;
while (p)
{
if (p&1) ans=(ll)ans*a%MOD;
a=(ll)a*a%MOD;p>>=1;
}
return ans;
}
#define inv(x) qpow(x,MOD-2)
inline int add(const int& x,const int& y){return x+y>=MOD? x+y-MOD:x+y;}
inline int dec(const int& x,const int& y){return x<y? x-y+MOD:x-y;}
int rt[2][24];
int r[MAXN],l,lim;
inline void init(){lim=1<<l;for (int i=0;i<lim;i++) r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));}
inline void NTT(int* a,int type)
{
for (int i=0;i<lim;i++) if (i<r[i]) swap(a[i],a[r[i]]);
for (int L=0;L<l;L++)
{
int mid=1<<L,len=mid<<1;
ll Wn=rt[type][L+1];
for (int s=0;s<lim;s+=len)
for (int k=0,w=1;k<mid;k++,w=w*Wn%MOD)
{
int x=a[s+k],y=(ll)w*a[s+mid+k]%MOD;
a[s+k]=add(x,y),a[s+mid+k]=dec(x,y);
}
}
if (type)
{
int t=inv(lim);
for (int i=0;i<lim;i++) a[i]=(ll)a[i]*t%MOD;
}
}
int fac[MAXN],finv[MAXN];
int a[MAXN];
int f[MAXN],g[MAXN];
int main()
{
rt[0][23]=qpow(3,119);rt[1][23]=inv(rt[0][23]);
for (int i=22;i>=0;i--)
{
rt[0][i]=(ll)rt[0][i+1]*rt[0][i+1]%MOD;
rt[1][i]=(ll)rt[1][i+1]*rt[1][i+1]%MOD;
}
int n=read();
ll m;
scanf("%lld",&m);
m%=MOD-1;
for (int i=0;i<=n;i++) a[i]=read();
fac[0]=1;
for (int i=1;i<=n;i++) fac[i]=(ll)fac[i-1]*i%MOD;
finv[n]=inv(fac[n]);
for (int i=n-1;i>=0;i--) finv[i]=(ll)finv[i+1]*(i+1)%MOD;
for (int i=0;i<=n;i++) f[i]=(ll)a[i]*fac[i]%MOD;
for (int i=0;i<=n;i++) g[i]=finv[n-i];
for (l=0;(1<<l)<=(n<<1);++l);
init();
NTT(f,0);NTT(g,0);
for (int i=0;i<lim;i++) f[i]=(ll)f[i]*g[i]%MOD;
NTT(f,1);
for (int i=0;i<=n;i++) a[i]=(ll)f[i+n]*finv[i]%MOD;
for (int i=0;i<=n;i++) a[i]=(ll)a[i]*inv(qpow(i+1,m))%MOD;
for (int i=0;i<=n;i++) f[i]=(ll)a[i]*fac[i]%MOD;
for (int i=0;i<=n;i++) g[i]=(((n-i)&1)? MOD-finv[n-i]:finv[n-i]);
for (int i=n+1;i<lim;i++) f[i]=g[i]=0;
NTT(f,0);NTT(g,0);
for (int i=0;i<lim;i++) f[i]=(ll)f[i]*g[i]%MOD;
NTT(f,1);
for (int i=0;i<=n;i++) a[i]=(ll)f[i+n]*finv[i]%MOD;
for (int i=0;i<=n;i++) printf("%d ",a[i]);
return 0;
}