【BZOJ 2510】弱题 循环矩阵

首先写dp方程，设f[i][j]表示第i次后第j中种1球的期望个数，所以

f[i][j]=f[i-1][j]+f[i-1][j-1]/m*1-f[i-1][j]/m*1+(1-(f[i-1][j]+f[i-1][j-1])/m)*0

->f[i][j]=f[i-1][j](1-1/m)+f[i-1][j-1]/m

矩阵加速，然后发现是循环矩阵，就只用储存第一行的信息，完事。

#include<cstdio>
#include<cstring>
#include<iostream>
#define maxn 1021
using namespace std;
int nu[maxn][maxn],n,m,k;
double mat[maxn],e[maxn],ans[maxn],f[maxn],F[maxn];
int last[maxn][2];
inline int Q(int x){
	if(x==0)return n;if(x==n+1)return 1;
	return x;
}
void make(){
	for(int i=1;i<=n;i++)last[i][0]=i;
	int pos=0;
	for(int i=1;i<=n;i++){
		pos^=1;
		for(int j=1;j<=n;j++){
			nu[i][j]=last[j][pos^1];
			last[j][pos]=last[Q(j-1)][pos^1];
		}
	}
}
void mul(double a[maxn],double b[maxn],double c[maxn]){
	for(int i=1;i<=n;i++)e[i]=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			e[i]+=a[j]*b[nu[j][i]];
		}
	}
	for(int i=1;i<=n;i++)c[i]=e[i];
}
void mull(int y){
	ans[1]=1;
	for(;y;y>>=1){
		if(y&1)mul(ans,mat,ans);
		mul(mat,mat,mat);
	}
}

void solve(){
	mat[1]=1-1.0/m;mat[n]=1.0/m;
	mull(k);
	
	for(int i=1;i<=n;i++){
		for(int j=1;j<=n;j++){
			F[i]+=ans[nu[i][j]]*f[j];
		}
	}
	for(int i=1;i<=n;i++)printf("%.3lf\n",F[i]);
}

int main(){
	scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;i++)scanf("%lf",f+i);
	make();
	solve();
	return 0;
}