luogu#P11420. [清华集训 2024] 乘积的期望

[清华集训 2024] 乘积的期望

题目描述

有一个长度为 nn 的序列 a1,a2,,ana_1, a_2, \cdots , a_n。初始序列的所有元素均为 00。再给定正整数 mmcc(nm+1)(n - m + 1) 个正整数 b1,b2,,bnm+1b_1, b_2, \cdots , b_{n−m+1}

对序列 a1,a2,,ana_1, a_2, \cdots , a_n 进行 cc 次操作,每次操作为:

  • 随机选择整数 1xnm+11 \le x \le n - m + 1,其中选到 yy1ynm+11\le y\le n-m+1)的概率为 byi=1nm+1bi\displaystyle \frac{b_y}{\sum_{i=1}^{n-m+1} b_i}。将 ax,ax+1,,ax+m1a_x, a_{x+1}, \cdots, a_{x+m−1} 增加 11

cc 次操作中对 xx 的随机是独立的。

求操作完成后序列中所有元素的乘积的期望。为了避免浮点数输出,你需要将答案对 998244353998244353 取模。

输入格式

从标准输入读入数据。

第一行三个整数 n,m,cn, m, c,分别表示序列长度、操作区间长度和操作次数。

第二行 (nm+1)(n - m + 1) 个整数 b1,,bnm+1b_1, \cdots , b_{n-m+1},描述随机的权重。

输出格式

输出到标准输出。

输出一行一个整数,表示 cc 次操作后序列中所有数的乘积的期望。

3 2 2
1 1

1

10 3 10
1 2 3 4 5 6 7 8
721023399
20 12 98765
9 8 7 6 5 4 3 2 1
560770686

提示

【样例 11 解释】

当两次操作选择的 xx 不同时,最终序列为 [1,2,1][1,2,1],序列元素乘积为 22;否则序列为 [2,2,0][2,2,0][0,2,2][0,2,2],序列元素乘积均为 00。两次操作选择的 xx 不同的概率为 12\frac{1}{2} ,因此输出 2×12=12\times \frac{1}{2}=1

子任务

对于所有测试数据,2mn502 \le m \le n \le 501c<9982443531 \le c < 998244353,对于 1inm+11 \le i \le n - m + 11bi1051 \le b_i \le 10^5

  • Subtask 1(10%10\%):m8m \le 8
  • Subtask 2(20%20\%):m16m \le 16
  • Subtask 3(15%15\%):n20,cnn \le 20, c \le n
  • Subtask 4(15%15\%):n30,cnn \le 30, c \le n
  • Subtask 5(15%15\%):n40,cnn \le 40, c \le n
  • Subtask 6(15%15\%):cnc \le n
  • Subtask 7(10%10\%):无特殊限制。