bzoj2510 弱题 循环矩阵

传送门

循环矩阵,学习到了。
这里解释得很详细

a [ x ] a[x] 表示编号为 i i 的期望。转移就是
a [ x ] = a [ x ] × m 1 m + a [ x 1 ] × 1 m a[x]=a[x]\times\frac{m-1}{m}+a[x-1]\times\frac{1}{m}

然后用矩阵优化。 O ( n 3   l o g k ) O(n^3\ logk) 过不了。用循环矩阵优化。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e3+10;
double m,num[maxn];
int n,k;
struct matrix{
	double a[maxn];
	matrix(){for(int i=0;i<maxn;++i) a[i]=0.0;}
	friend inline matrix operator*(const matrix &A,const matrix &B){
		matrix ret;
		for(int i=0;i<n;++i) for(int j=0;j<n;++j)
			ret.a[(i+j)%n]+=A.a[i]*B.a[j];
		return ret;	
	}
	friend inline matrix operator^(matrix &A,int b){
		matrix ret;ret.a[0]=1.0;
		for(;b;A=A*A,b>>=1) if(b&1) ret=ret*A;
		return ret;
	}
}A,B;
int main(){
	while(scanf("%d%lf%d",&n,&m,&k)!=EOF){
		for(int i=0;i<n;++i) scanf("%lf",&A.a[i]);
		B.a[0]=1.0-1.0/m,B.a[1]=1.0/m,A=A*(B^k);
		for(int i=0;i<n;++i) printf("%.3lf\n",A.a[i]);
	}
}

猜你喜欢

转载自blog.csdn.net/g21wcr/article/details/88592904