求k时刻一个标号转移到各位置的概率,最后枚举每个标号加权求期望。可以发现转移矩阵是循环矩阵,因此乘法是n^2的。另外这个乘法是圆周卷积的形式,然后就作死写了发fft,发现精度升天了= =
#include<cstdio> #include<cstring> #define N 1000 int n,m,k,i,j; typedef double ds[N]; ds s,t,u,q; void mul(double* s,double* t){ memset(u,0,n*8); for(i=0;i!=n;++i) for(j=0;j!=n;++j) u[(i+j)%n]+=s[i]*t[j]; memcpy(s,u,n*8); } int main(){ scanf("%d%d%d",&n,&m,&k); for(i=0;i!=n;++i) scanf("%lf",t+i); *q=1-1./m; q[1]=1./m; for(*s=1;k;k/=2){ if(k%2)mul(s,q); if(k/2)mul(q,q); else mul(t,s); } for(i=0;i!=n;++i) printf("%.3f\n",t[i]); }
时间: 2024-10-12 08:08:32