BZOJ2510 弱题

设a[x]为编号为x的球的期望个数，每进行一次操作时候编号为x的球的期望个数就变成了a[x]-a[x]/m+a[x-1]/m，这个东西显然可以构造一个矩阵来转移

就这样进行矩阵快速幂的话是n^3*log INF的，由于n=1000过不了（为什么没有n=100,k=INFMAX的部分分-_-）

但是我们发现这个矩阵不管自乘多少次第i行都是第i-1行右移一位，第i列都是第i-1列下移一位，而第一列是把第一行绕（1,1）逆时针转九十度

所以我们只需要维护第一行是什么就可以了，知道一行就可以知道整个矩阵的形态，而由于只需要维护一行，进行一次乘法是n^2的

复杂度O(n^2 log INF)

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<stack>
#include<queue>
#include<map>
#include<set>
#include<bitset>
using namespace std;
#define MAXN 1010
#define MAXM 1010
#define ll long long
#define INF 1000000000
#define MOD 1000000007
#define eps 1e-8
int n;
struct mat{
	double x[MAXN];
	mat(){
		int i;
		for(i=0;i<MAXN;i++){
			x[i]=0;
		}
	}
	friend mat operator *(mat x,mat y){
		mat z;
		int i,j;
		for(i=0;i<n;i++){
			for(j=0;j<n;j++){
				z.x[i]+=x.x[j]*(y.x[(i-j+n)%n]);
			}
		}
		return z;
	}
};
int m,k;
mat a,b,c;
int main(){
	int i;
	scanf("%d%d%d",&n,&m,&k);
	if(n==1){
		printf("%.3lf\n",1.0*m);
		return 0;
	}
	for(i=1;i<=n;i++){
		scanf("%lf",&a.x[i-1]);
	}
	b.x[0]=1;
	c.x[0]=-1.0/m+1;
	c.x[1]=1.0/m;
	while(k){
		if(k&1){
			b=b*c;
		}
		c=c*c;
		k>>=1;
	}
	a=a*b;
	for(i=0;i<n;i++){
		printf("%.3lf\n",a.x[i]);
	}
	return 0;
}

/*
2 3 2
3 0
*/