[bzoj2510] 弱题 （dp+循环矩阵）

Description

有$M$个球，一开始每个球均有一个初始标号，标号范围为$1$～$N$且为整数，标号为i的球有$a_i$个，并保证$Σa_i = M$。
每次操作等概率取出一个球（即取出每个球的概率均为$1/M$），若这个球标号为$k（k < N）$，则将它重新标号为$k + 1$；若这个球标号为$N$，则将其重标号为$1$。（取出球后并不将其丢弃）
现在你需要求出，经过K次这样的操作后，每个标号的球的期望个数。

Input

第1行包含三个正整数$N，M，K$，表示了标号与球的个数以及操作次数。
第2行包含$N$个非负整数$a_i$，表示初始标号为i的球有$a_i$个。

Output

应包含$N$行，第i行为标号为i的球的期望个数，四舍五入保留$3$位小数。

Sample Input

2 3 2
3 0

Sample Output

1.667
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$。

题解

首先感谢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;
}