BZOJ2510 弱题

---

Address

• BZOJ2510

---

Solution

• 设 $f[i][j]$ 表示第 $i$ 次操作后，标号为 $j$ 的球的期望个数。
• 记 $pre[i] = i - 1(2 \le i \le n), pre[1] = n$。
• 则若第 $i - 1$ 次操作后，标号为 $j$ 的球的期望个数为 $x$，标号为 $pre[j]$ 的球的期望个数为 $y$，对于第 $i$ 次操作：
  
  • 有 $\frac{x}{m}$ 的概率取出标号为 $j$ 的球，操作后标号为 $j$ 的个数变为 $x - 1$。
  • 有 $\frac{y}{m}$ 的概率取出标号为 $pre[j]$ 的球，操作后标号为 $j$ 的个数变为 $x + 1$。
  • 有 $\frac{m - x - y}{m}$ 的概率取出其它球，操作后标号为 $j$ 的个数不变。
• 因此可以得到转移方程： $$f[i][j]=\frac{f[i-1][pre[j]]}{m}\times(f[i-1][j]+1)+\frac{f[i-1][j]}{m}\times(f[i-1][j]-1)+\frac{m-f[i-1][pre[j]]-f[i-1][j]}{m}\times f[i-1][j]$$
• 展开消去同类项，得到 $f[i][j] = \frac{m - 1}{m}f[i - 1][j] + \frac{1}{m}f[i - 1][pre[j]]$。
• 时间复杂度 $O(nK)$，显然不能通过。
• 考虑矩阵优化，复杂度就被优化到了 $O(n^3\log K)$，转移矩阵如下：
  $$\left\{ \begin{matrix} \frac{m - 1}{m} & \frac{1}{m} & 0 & \cdots & 0 \\ 0 & \frac{m - 1}{m} & \frac{1}{m} & \cdots & 0 \\ 0 & 0 & \frac{m - 1}{m} & \cdots & 0 \\ \vdots & \vdots &\vdots &\ddots & \vdots \\ \frac{1}{m} & 0 & 0 & \cdots & \frac{1}{m} \\ \end{matrix} \right\}$$
• 尽管依然不能通过，我们会发现转移矩阵的任意次方都是这样的形式：
  $$\left\{ \begin{matrix} x_1&x_2&x_3&...&x_n \\ x_n&x_1&x_2&...&x_{n-1} \\ x_{n-1}&x_n&x_1&...&x_{n-2}\\ \vdots &\vdots &\vdots &\ddots & \vdots \\x_2&x_3&x_4&...&x_1 \end{matrix} \right\}$$
• 因此只要计算出矩阵第一行的值，其余的行都可以随之确定，时间复杂度 $O(n^2 \log K)$。

---

Code

• 仿佛这里的矩阵由于空间太大不能用结构体写

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <cctype>

using namespace std;

const int N = 1005;
int n, K, pre[N]; double m;
double c[N][N], a[N], b[N][N], d[N][N], ans[N];

inline void Push(double x[N][N], double y[N][N])
{
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= n; ++j)
            x[i][j] = y[i][j];
}

inline void Join(double x[N][N], double y[N][N], double z[N][N])
{
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= n; ++j)
            z[i][j] = 0.0;
    for (int k = 1; k <= n; ++k)
        for (int j = 1; j <= n; ++j)
            z[1][j] += x[1][k] * y[k][j];
    for (int i = 2; i <= n; ++i)
        for (int j = 1; j <= n; ++j)
            z[i][j] = z[i - 1][pre[j]];
}

int main()
{
    scanf("%d%lf%d", &n, &m, &K);
    for (int i = 1; i <= n; ++i) 
        scanf("%lf", &a[i]);
    for (int i = 2; i <= n; ++i) 
        pre[i] = i - 1; pre[1] = n;
    double p1 = (m - 1.0) / m, p2 = 1.0 / m;
    for (int j = 1; j <= n; ++j) 
        b[j][j] = p1, b[pre[j]][j] = p2;

    for (int i = 1; i <= n; ++i) c[i][i] = 1;
    while (K)
    {
        if (K & 1)
            Join(b, c, d), Push(c, d);
        Join(b, b, d); Push(b, d); K >>= 1;
    }
    for (int i = 1; i <= n; ++i)
        for (int j = 1; j <= n; ++j)
            ans[j] += a[i] * c[i][j];

    for (int i = 1; i <= n; ++i)
        printf("%.3lf\n", ans[i]);
    return 0;
}