Description
有
M
M
个球,一开始每个球均有一个初始标号,标号范围为~
N
N
且为整数,标号为i的球有个,并保证
Σai=M
Σ
a
i
=
M
。
每次操作等概率取出一个球(即取出每个球的概率均为
1/M
1
/
M
),若这个球标号为
k(k<N)
k
(
k
<
N
)
,则将它重新标号为
k+1
k
+
1
;若这个球标号为
N
N
,则将其重标号为。(取出球后并不将其丢弃)
现在你需要求出,经过K次这样的操作后,每个标号的球的期望个数。
Input
第1行包含三个正整数
N,M,K
N
,
M
,
K
,表示了标号与球的个数以及操作次数。
第2行包含
N
N
个非负整数,表示初始标号为i的球有
ai
a
i
个。
Output
应包含 N N 行,第i行为标号为i的球的期望个数,四舍五入保留位小数。
Sample Input
2 3 2
3 0
Sample Output
1.667
1.333
HINT
【样例说明】
第 1 1 次操作后,由于标号为球个数为 0 0 ,所以必然是一个标号为的球变为标号为 2 2 的球。所以有个标号为 1 1 的球,有个标号为 2 2 的球。
第次操作后,有 1/3 1 / 3 的概率标号为 2 2 的球变为标号为的球(此时标号为 1 1 的球有个),有 2/3 2 / 3 的概率标号为 1 1 的球变为标号为的球(此时标号为 1 1 的球有个),所以标号为 1 1 的球的期望个数为。同理可求出标号为 2 2 的球期望个数为。
【数据规模与约定】
对于 10 10 %的数据, N≤5,M≤5,K≤10 N ≤ 5 , M ≤ 5 , K ≤ 10 ;
对于 20 20 %的数据, N≤20,M≤50,K≤20 N ≤ 20 , M ≤ 50 , K ≤ 20 ;
对于 30 30 %的数据, N≤100,M≤100,K≤100 N ≤ 100 , M ≤ 100 , K ≤ 100 ;
对于 40 40 %的数据, M≤1000,K≤1000 M ≤ 1000 , K ≤ 1000 ;
对于 100 100 %的数据, N≤1000,M≤100,000,000,K≤2,147,483,647 N ≤ 1000 , M ≤ 100 , 000 , 000 , K ≤ 2 , 147 , 483 , 647 。
题解
首先感谢lvmaomao的讲解,贴一波链接(点蓝字)
再贴一篇写得也很好的博客,点这里。
这题的主要技巧就是循环矩阵,具体实现可以参考上面两篇博客。
Code
#include<bits/stdc++.h>
using namespace std;
int n,m,k;
double fin[1010][1010],ans[1010];
struct matix{
double b[1010];
matix operator * (const matix &B)const{
matix C;
for(int i=1;i<=n;i++)C.b[i]=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
C.b[(i+j-2)%n+1]+=b[i]*B.b[j];
return C;
}
}A,B,t;
int main()
{
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)scanf("%lf",&A.b[i]);
B.b[1]=(double)(m-1)/(double)m;
B.b[2]=1.0/(double)m;
t.b[1]=1;
while(k){
if(k&1)t=t*B;
B=B*B;
k>>=1;
}
A=A*t;
for(int i=1;i<=n;i++)printf("%.3lf\n",A.b[i]);
return 0;
}