题意
有n只鸽子,每个时刻会等概率随机给某只鸽子喂食,若一只鸽子被喂食的次数不小于k次则已饱。问期望多少个时刻之后每只鸽子都已饱。
n
≤
50
,
k
≤
1000
n\le50,k\le1000
n≤50,k≤1000
分析
先min-max容斥一下转成求
E
(
m
i
n
(
t
i
)
)
E(min(t_i))
E(min(ti))。
然后问题变成求前m个盒子恰好有一个被放了k个球的期望时间。
设这m个盒子总共放了x个球,那么到达某个合法状态的期望时间就是
x
m
n
\frac{x}{\frac{m}{n}}
nmx,概率是
1
m
x
\frac{1}{m^x}
mx1,这就变成了一个计数题,用FFT优化一下就好。
时间复杂度
O
(
n
2
k
l
o
g
(
n
k
)
)
O(n^2klog(nk))
O(n2klog(nk))。
代码
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
typedef long long LL;
const int N=55;
const int M=1005;
const int MOD=998244353;
int n,m,f[N][N*M],jc[N*M],ny[N*M],a[N*M*2],b[N*M*2],T,rev[N*M*2];
int ksm(int x,int y)
{
int ans=1;
while (y)
{
if (y&1) ans=(LL)ans*x%MOD;
x=(LL)x*x%MOD;y>>=1;
}
return ans;
}
void NTT(int *a,int f)
{
for (int i=0;i<T;i++) if (i<rev[i]) std::swap(a[i],a[rev[i]]);
for (int i=1;i<T;i<<=1)
{
int wn=ksm(3,f==1?(MOD-1)/i/2:MOD-1-(MOD-1)/i/2);
for (int j=0;j<T;j+=i*2)
{
int w=1;
for (int k=0;k<i;k++)
{
int u=a[j+k],v=(LL)a[j+k+i]*w%MOD;
a[j+k]=(u+v)%MOD;a[j+k+i]=(u-v)%MOD;
w=(LL)w*wn%MOD;
}
}
}
int t=ksm(T,MOD-2);
if (f==-1) for (int i=0;i<T;i++) a[i]=(LL)a[i]*t%MOD;
}
void work()
{
f[0][0]=1;
for (int i=0;i<m;i++) f[1][i]=ny[i];
int now=m-1;
for (int i=2;i<=n;i++)
{
int lg=0;
for (T=1;T<=now+m-1;T<<=1,lg++);
for (int j=0;j<T;j++) rev[j]=(rev[j>>1]>>1)|((j&1)<<(lg-1)),a[j]=b[j]=0;
for (int j=0;j<=now;j++) a[j]=f[i-1][j];
for (int j=0;j<m;j++) b[j]=ny[j];
NTT(b,1);NTT(a,1);
for (int j=0;j<T;j++) a[j]=(LL)b[j]*a[j]%MOD;
NTT(a,-1);
now+=m-1;
for (int j=0;j<=now;j++) f[i][j]=a[j];
}
}
int main()
{
scanf("%d%d",&n,&m);
jc[0]=jc[1]=ny[0]=ny[1]=1;
for (int i=2;i<=n*m;i++) jc[i]=(LL)jc[i-1]*i%MOD,ny[i]=(LL)(MOD-MOD/i)*ny[MOD%i]%MOD;
for (int i=2;i<=n*m;i++) ny[i]=(LL)ny[i-1]*ny[i]%MOD;
work();
int ans=0;
for (int i=1;i<=n;i++)
{
int tot=0,w=0,t=ksm(i,MOD-2),h=ksm(t,m);
for (int j=m;j<=i*(m-1)+1;j++)
{
(w+=(LL)f[i-1][j-m]*i%MOD*ny[m-1]%MOD*jc[j-1]%MOD*n%MOD*j%MOD*t%MOD*h%MOD)%=MOD;
h=(LL)h*t%MOD;
}
w=(LL)w*jc[n]%MOD*ny[i]%MOD*ny[n-i]%MOD;
if (i&1) (ans+=w)%=MOD;
else (ans-=w)%=MOD;
}
ans+=ans<0?MOD:0;
printf("%d\n",ans);
return 0;
}