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
3 0
Sample Output
1.667
1.333
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
Source
2011福建集训
题解
矩阵乘法+快速幂。也是第一题关于期望的。
对于每次操作,编号为i的球的个数只取决于上一轮编号为i-1和i的球:编号为i的球每个都有1/m的几率被抽中,使之编号变大;编号为i-1的球每个都有(m-1)/m的几率不会被抽中,使之编号不变。
据此可列矩阵乘法,然而作为底数的矩阵(我更习惯称之为基础矩阵)是循环矩阵(理解上一段后自行画图证明)。于是只需记第一行即可。
具体见代码。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
using namespace std;
int n,m,k;
double a[1002]/*基准矩阵*/,b[1002]/*对角矩阵*/,ans[1002];
void mul(double a[1002],double b[1002],double c[1002])
{
double t[1002];
for(int i=1;i<=n;i++)
{t[i]=0;
for(int j=1;j<=n;j++) t[i]+=b[j]*a[(n+i-j)%n+1];
}
for(int i=1;i<=n;i++) c[i]=t[i];
}
void ksm()
{
b[1]=1; a[1]=(double)(m-1)/m; a[n]=(double)1/m;
while(k)
{if(k&1) mul(a,b,b);
mul(a,a,a); k=k>>1;
}
}
void cheng(double a[1002],double b[1002],double c[1002])
{
double t[1002];
for(int i=1;i<=n;i++)
{t[i]=0;
for(int j=1;j<=n;j++) t[i]+=a[(n+j-i)%n+1]*b[j];
}
for(int i=1;i<=n;i++) c[i]=t[i];
}
int main()
{
scanf("%d%d%d",&n,&m,&k);
ksm();
for(int i=1;i<=n;i++) scanf("%lf",&ans[i]);
cheng(b,ans,ans);
for(int i=1;i<=n;i++) printf("%.3lf\n",ans[i]);
return 0;
}