2022牛客五一集训派对day2 E.Music Game

题面

题意

• 给定长度为 $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$

思路

1. 暴力 $O(n^2)$ 求所有连续区间的概率，注意一位为 1 的概率 0 会影响左右两段，注意讨论，求前缀积的时候跳过这样的位。区间连续 1 的概率即为 $s_i/s_{j-1}$，如果 $s_{j-1}$ 为 0，说明 $s_i$ 从第 $j$ 位开始前缀积，此时不需要除 $s_{j-1}$，否则要除。

2. 第 $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})$。

3. 注意细节：

   if(a[j] == 0) break;
   

   不然，i~j 的段中间可能会有 0，无法保证 i~j 都是 1

4. 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;
}