BZOJ2510 弱题

Description

　　有M个球，一开始每个球均有一个初始标号，标号范围为1～N且为整数，标号为i的球有ai个，并保证Σai = M。
　　每次操作等概率取出一个球（即取出每个球的概率均为1/M），若这个球标号为k（k < N），则将它重新标号为k + 1；若这个球标号为N，则将其重标号为1。（取出球后并不将其丢弃）
　　现在你需要求出，经过K次这样的操作后，每个标号的球的期望个数。

Input

　　第1行包含三个正整数N，M，K，表示了标号与球的个数以及操作次数。
　　第2行包含N个非负整数ai，表示初始标号为i的球有ai个。

Output

　　输出应包含N行，第i行为标号为i的球的期望个数，四舍五入保留3位小数。

Sample Input

2 3 2

3 0

Sample Output

1.667

1.333

Hint

【样例说明】
　　第1次操作后，由于标号为2球个数为0，所以必然是一个标号为1的球变为标号为2的球。所以有2个标号为1的球，有1个标号为2的球。
　　第2次操作后，有1/3的概率标号为2的球变为标号为1的球（此时标号为1的球有3个），有2/3的概率标号为1的球变为标号为2的球（此时标号为1的球有1个），所以标号为1的球的期望个数为1/3*3+2/3*1 = 5/3。同理可求出标号为2的球期望个数为4/3。
【数据规模与约定】
　　对于10%的数据，N ≤ 5, M ≤ 5, K ≤ 10；
　　对于20%的数据，N ≤ 20, M ≤ 50, K ≤ 20；
　　对于30%的数据，N ≤ 100, M ≤ 100, K ≤ 100；
　　对于40%的数据，M ≤ 1000, K ≤ 1000；
　　对于100%的数据，N ≤ 1000, M ≤ 100,000,000, K ≤ 2,147,483,647。

首先明确方程：f[i][j]=f[i-1][j-1]*1/m+f[i-1][j]*(1-1/m)

看到k非常大，先打表看一波规律，发现什么都没有；于是想到矩阵快速幂优化，又发现1000^3超时，然后就弃疗= =

然而，该矩阵是可以优化的！！！

发现矩阵B是个循环矩阵，任意一行可由另外一行表示出来= =

那么B'=B*B=(这个就是找规律了)，于是一次矩阵乘法降到了n^2，时间复杂度n^2logk

#include<bits/stdc++.h>
using namespace std;
const int Maxn=1005;
double f[Maxn],tmp[Maxn];
struct Matrix{
	double mat[Maxn];
	inline void init(int x,int siz){
		for(int i=1;i<=siz;++i)mat[i]=x;
	}
};
inline Matrix mult(Matrix a,Matrix b,int n){
	Matrix c;c.init(0,n);
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			c.mat[(i+j-2)%n+1]+=a.mat[i]*b.mat[j];
	return c;
}
inline Matrix ksm(Matrix a,int b,int n){
	Matrix ret=a;
	Matrix ans;ans.init(0,n);ans.mat[1]=1;
	for(;b;b>>=1){
		if(b&1)ans=mult(ans,ret,n);
		ret=mult(ret,ret,n);
	}
	return ans;
}
int main(){
	int n,m,k;scanf("%d%d%d",&n,&m,&k);
	for(int i=1;i<=n;++i)scanf("%lf",&tmp[i]);
	Matrix b;b.init(0,n);
	b.mat[1]=(m-1)/(m-0.0),b.mat[2]=1/(m-0.0);
	b=ksm(b,k,n);
	for(int i=1;i<=n;++i)
		for(int j=1;j<=n;++j)
			f[(i+j-2)%n+1]+=tmp[i]*b.mat[j];
	for(int i=1;i<=n;++i)cout<<fixed<<setprecision(3)<<f[i]<<'\n';
	return 0;
}