2022牛客五一集训派对day2 E.Music Game_牛客五一集训2-CSDN博客

本文链接：https://blog.csdn.net/aldolreaction/article/details/124534721

题意

给定长度为 $n$ 的 01 序列每一位为 1 的概率 $p_i$ ，给定 $m$ ，长度为 $x$ 的连续一段对答案的贡献是 $x^m$ ，求答案在模意义下的期望。
$1\le n\le 1000, 1\le m\le 1000, 0\le p[i]\le 100$

思路

暴力 $O(n^2)$ 求所有连续区间的概率，注意一位为 1 的概率 0 会影响左右两段，注意讨论，求前缀积的时候跳过这样的位。区间连续 1 的概率即为 $s_i /s_{j-1}$ ，如果 $s_{j-1}$ 为 0，说明 $s_i$ 从第 $j$ 位开始前缀积，此时不需要除 $s_{j-1}$ ，否则要除。
第 $i - 1, j + 1$ 位必须要为 0，才是第 $i$ 到 $j$ 为连续 1 段的贡献。要乘上这两位为 0 的概率 $1-a_{i-1}),(1-a_{j+1})$ 。如果 $i$ 是第一位，前面一位不需要有零，不要乘 $1-a_{i-1})$ 。同理，如果 $i$ 是最后一位，后面一位不需要有零，不要乘 $1-a_{j+1})$ 。
注意细节：
```
if(a[j] == 0) break;
```
不然，i~j 的段中间可能会有 0，无法保证 i~j 都是 1
mint 为自动取模的整数类：https://paste.ubuntu.com/p/KyGhdcSc6j/

代码

int n, m;
mint ans, a[N], s[N];

int main() {
    cin >> n >> m;
    a[0] = s[0] = 1;
    rep(i, 1, n) cin >> a[i], a[i] /= 100, s[i] = a[i];
    rep(i, 1, n) {
        if(a[i] == 0 or a[i - 1] == 0) continue;
        s[i] *= s[i - 1];
    }
    rep(i, 1, n) {
        if(a[i] == 0) continue;
        rep(j, i, n) {
            if(a[j] == 0) break;
            mint dx = qpow(j - i + 1, m) * s[j];
            if(a[i - 1] != 0) {
                dx /= s[i - 1];
                if(i != 1) dx *= (1 - a[i - 1]);
            }
            if(a[j + 1] != 0 and j != n) dx *= (1 - a[j + 1]);
            ans += dx;
        }
    }
    cout << ans << endl;
    return 0;
}