BZOJ2510 弱题/取球游戏

题目大意：

有M个球，一开始每个球均有一个初始标号，标号范围为1～N且为整数，标号为i的球有a_i个，并保证Σa_i = M。

每次操作等概率取出一个球（即取出每个球的概率均为1/M），若这个球标号为k（k < N），则将它重新标号为k + 1；若这个球标号为N，则将其重标号为1。（取出球后并不将其丢弃）

现在你需要求出，经过K次这样的操作后，每个标号的球的期望个数。

题目分析：先考虑对于第i个球的期望个数，设为a[i]。则a[i]=(a[i])/m*(a[i]-1)+a[i-1]/m*(a[i]+1)+(m-a[i]-a[i-1])/m*a[i]。

两个加号隔开的三部分分别表示第i号球被选和第i-1号球被选以及两个球都没被选。

化简：a[i]=(a[i]*(a[i]-1)+a[i-1]*a[i]+a[i-1]+(m-a[i]-a[i-1])*a[i])/m

                =((a[i]-1+a[i-1]+m-a[i]-a[i-1])*a[i]+a[i-1])/m

                =((m-1)*a[i]+a[i-1])/m

                =(m-1)/m*a[i]+a[i-1]/m

由化简结果得知a[i]与a[i-1]和m有关系，可以构造出一个如下的矩阵

由于n<=1000所以不能用n三方log(k)的做法。

观察可知这个矩阵是一个循环矩阵,即A[i][j]=A[i+1][j+1];

所以得知第一行或者第一列就可以推出其余的矩阵。

对于矩阵a[i][j]有a[i][j]=sigma(a[i][k]*b[k][j])

只考虑第一行所以：a[i]=sigma(a[j]*b[j][i]);->省去一维。

总时间复杂度为O(n²log(k))。

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
#include<cmath>
#include<cctype>
#include<cassert>
#include<climits>
using namespace std;
#define For(i,n) for(int i=1;i<=n;i++)
#define Rep(i,n) for(int i=0;i<n;i++)
#define Fork(i,k,n) for(int i=k;i<=n;i++)
#define ForD(i,n) for(int i=n;i;i--)
#define Forp(x) for(int p=pre[x];p;p=next[p])
#define RepD(i,n) for(int i=n;i>=0;i--)
#define MEM(a) memset(a,0,sizeof(a))
#define MEMI(a) memset(a,127,sizeof(a))
#define MEMi(a) memset(a,128,sizeof(a))
#define INF (2139062143)
#define phiF (1000000006)
#define MAXN (1000000+10)
typedef long long LL;

int n,m,k;
double c[1005],a[1005],b[1005];

int main(){
	scanf("%d%d%d",&n,&m,&k);
	a[1]=(m-1.0)/m;a[2]=1.0/m;
	b[1]=1;
	while (k){
		if (k%2) {
			MEM(c);
			For (i,n) 
			 For (j,n)
			  c[i]+=a[j]*b[(i-j+n)%n+1];
			For (i,n)
			 b[i]=c[i];
		}
		k>>=1;
		MEM(c);
		For (i,n) 
		 For (j,n)
		  c[i]+=a[j]*a[(i-j+n)%n+1];
		For (i,n)
		 a[i]=c[i];
	}
	
	For (i,n)
	 scanf("%lf",&a[i]);
	MEM(c);
	For (i,n)
	 For (j,n)
	  c[i]+=a[j]*b[(i-j+n)%n+1];
	For (i,n)
	 printf("%.3f\n",c[i]);
}