Bzoj2510 弱题(矩阵快速幂)

Posted water-mi

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Bzoj2510 弱题(矩阵快速幂)相关的知识,希望对你有一定的参考价值。

题面(权限题)

题解

一道概率(dp),可以设(f[i][j])表示第(i)次操作后,标号为(j)的小球的期望个数,那么有:
[ egin{aligned} &f[i][j]=(1-frac 1m)f[i-1][j]+frac1mf[i-1][j-1](1leq jleq n) &f[i][0]=(1-frac 1m)f[i-1][j]+frac1mf[i-1][n] end{aligned} ]
这样的话转移可以写成矩阵的形式(假设有(4)个小球):
[ egin{aligned} &egin{bmatrix} f[i-1][1]&f[i-1][2]&f[i-1][3]&f[i-1][4] end{bmatrix} imes egin{bmatrix} 1-frac 1m&frac 1m&0&0&1-frac 1m&frac 1m&0&0&1-frac 1m&frac 1m\frac 1m&0&0&1-frac 1m end{bmatrix} \= &egin{bmatrix} f[i][1]&f[i][2]&f[i][3]&f[i][4] end{bmatrix} end{aligned} ]
可以发现转移矩阵也是一个循环矩阵,也就是说,可以(O(n^2log_2k))做。

#include <cstdio>
#include <cstring>
#include <algorithm>
using std::min; using std::max;
using std::swap; using std::sort;
typedef long long ll;

template<typename T>
void read(T &x) {
    int flag = 1; x = 0; char ch = getchar();
    while(ch < '0' || ch > '9') { if(ch == '-') flag = -flag; ch = getchar(); }
    while(ch >= '0' && ch <= '9') x = x * 10 + ch - '0', ch = getchar(); x *= flag;
}

const int N = 1e3 + 10;
int n, m, k; double S[N], T[N], tmp[N];

void mul(double S[], double T[]) {
    memset(tmp, 0, sizeof tmp);
    for(int i = 1; i <= n; ++i)
        for(int j = 1; j <= n; ++j)
            tmp[(i + j - 2) % n + 1] += S[i] * T[j];
    memcpy(S, tmp, sizeof tmp);
}

int main () {
    read(n), read(m), read(k);
    for(int i = 1; i <= n; ++i) scanf("%lf", S + i);
    T[1] = 1 - 1.0 / m, T[2]= 1.0 / m;
    for(; k; k >>= 1, mul(T, T)) if(k & 1) mul(S, T);
    for(int i = 1; i <= n; ++i) printf("%.3lf
", S[i]);
    return 0;
} 

以上是关于Bzoj2510 弱题(矩阵快速幂)的主要内容,如果未能解决你的问题,请参考以下文章

bzoj2510: 弱题

bzoj2875随机数生成器——矩阵快速幂

BZOJ 4000: [TJOI2015]棋盘( 状压dp + 矩阵快速幂 )

bzoj1875 [SDOI2009]HH去散步——矩阵快速幂

BZOJ 1409 Password 数论(扩展欧拉+矩阵快速幂+快速幂)

BZOJ2004公交线路(动态规划,状态压缩,矩阵快速幂)