luogu#P7509. 撕裂抹消

撕裂抹消

题目背景

何以撕裂,何以抹消?

题目描述

给定一个长为 nn 的整数序列 a1,a2,,ana_1,a_2,\dots,a_n,再给定一个有理数序列 p1,p2,,pnp_1,p_2,\dots,p_n。每个位置上有一个随机系数 ci{0,1}c_i \in \{0,1\},其中 P(ci=1)=piP(c_i = 1) = p_iP(ci=0)=1piP(c_i = 0) = 1-p_i

注意到随机系数写作序列后可以划分为若干个极长的连续 0011 段,极长指不能再向两边扩展。定义一种系数序列的权值为:当系数序列中恰有 kk 个极长连续 11 段时为 i=1nciai\sum\limits_{i=1}^n c_i a_i,否则为 00

求这个系数序列的期望权值。答案对 998244353998244353 取模。

输入格式

第一行,两个正整数 n,kn,k

第二行,nn 个非负整数 a1,a2,,ana_1,a_2,\dots,a_n

第三行,nn 个非负整数 p1,p2,,pnp_1,p_2,\dots,p_n,以模 998244353998244353 意义下给出。

输出格式

一行,一个非负整数,表示期望。

5 1
1 2 3 4 5
1 1 1 1 1
15
3 2
1 2 3
499122177 499122177 499122177
499122177

提示

样例解释

对于第一组样例,cic_i 必然为 11,且必然恰有 11 个极长连续段,故权值必然为 1+2+3+4+5=151 + 2 + 3 + 4 + 5 = 15

对于第二组样例,仅有 c1=1,c2=0,c3=1c_1 = 1,c_2 = 0,c_3 = 1 时权值不为 00,且此种情况的概率为 $\frac 12 \times \frac 12 \times \frac 12 = \frac 18$,故期望为 $\frac{1+3}8 = \frac 12 \equiv 499122177 \pmod {998244353}$。

数据范围

对于 30%30\% 的数据,n20n \le 20

对于 60%60\% 的数据,n103n \le 10^3

对于 100%100\% 的数据,1n1051 \le n \le 10^5,$1 \le k \le \min\left(20,\left\lfloor\frac{n+1}2\right\rfloor\right)$,0ai,pi<9982443530 \le a_i,p_i < 998244353