牛客OI周赛8-提高组 C 随机采矿

出题人绝对星际玩多了。

设表示时刻拥有个SCV的概率，可以得到：

 

那么时刻生产的晶体矿数量期望为：

当数据量较小时，可以直接写两重循环处理。当数据较大时，需要用矩阵加速。

其中，

会发现会不断变化，而实际上，在时，最多只有种可能性，因此可以分段处理，将相同的情况视为同一段。

若当前段的起始点为，则这一段的终点为，易证明当时，为定值。初始。

#include<bits/stdc++.h>
using namespace std;

const int MAXN=25;
int n,m;

struct Matrix
{
	double a[MAXN][MAXN];
	Matrix operator *(const Matrix m1)const
	{
		Matrix ret;
		memset(ret.a,0,sizeof(ret.a));
		for(int i=0;i<MAXN;i++)
		{
			for(int k=0;k<MAXN;k++)
			{
				if(a[i][k]==0)
					continue;
				for(int j=0;j<MAXN;j++)
					ret.a[i][j]+=a[i][k]*m1.a[k][j];
			}
		}
		return ret;
	}
};

Matrix powmod(Matrix m,long long p)
{
	Matrix ret;
	memset(ret.a,0,sizeof(ret.a));
	for(int i=0;i<MAXN;i++)
		ret.a[i][i]=1;
	while(p)
	{
		if(p&1)
			ret=ret*m;
		m=m*m;
		p>>=1;
	}
	return ret;
}

Matrix init(double p1)
{
	Matrix ret;
	memset(ret.a,0,sizeof(ret.a));
	for(int i=0;i<=m-n;i++)
	{
		ret.a[i][i]=1-p1;
		ret.a[i+1][i]=p1;
	}
	ret.a[m-n][m-n]=1;
	for(int i=0;i<=m-n;i++)
		ret.a[m-n+1][i]=n+i;
	ret.a[m-n+1][m-n+1]=1; 
	return ret;
}

int main()
{
	int T,k,i,ed;
	Matrix m1,ans;
	scanf("%d%d%d%d",&n,&m,&T,&k);
	memset(ans.a,0,sizeof(ans.a));
	ans.a[0][0]=1;
	for(i=1;i<=T;i=ed+1)
	{
		ed=T/(T/i);
		m1=init(T/i*1.0/T);
		ans=powmod(m1,ed-i+1)*ans;
	}
	printf("%.9lf\n",ans.a[m-n+1][0]*k);
}