#G. 暴力冲才是最优解

    传统题 1500ms 256MiB

暴力冲才是最优解

该比赛已结束,您无法在比赛模式下递交该题目。您可以点击“在题库中打开”以普通模式查看和递交本题。

题目描述

给你一个长度是$n$的并且数组中只含$0,1,2$的数组,让你求这个等式为多少$\sum_{i_1 = 1}^n\sum_{i_2 = i_1 + 1}^n\sum_{i_3 = i_2 + 1}^n{a_{i_1}}\times{a_{i_2}}\times{a_{i_3}}$小Y一看觉得这题觉得这个对于学弟们来说太easy了,一点挑战性都没有于是他加深了难度;

再给你一个整数$k$,让你求$\sum_{i_1 = 1}^n\sum_{i_2 = i_1 + 1}^n...\sum_{i_k = i_{k-1} + 1}^n{a_{i_1}}\times{a_{i_2}}\times...\times{a_{i_k}}$

你能求出上面这个式子的结果嘛,由于答案可能很大,你需要对$998244353$取模

输入格式

第一行两个整数$n ,k$ $( 1 ≤ n ≤ 5*10^6, 1 ≤ k ≤ n)$

第二行$n$个整数$a_1, a_2, a_3,...,a_n$ $(0 ≤ a_i ≤ 2)$

输出格式

输出一行一个整数, 并对$998244353$取模

样例

5 3
1 2 2 1 0
12

来源

2022 HGNU-SWUT暑假联合集训

HGNU ACM Training Round #11

未参加
状态
已结束
规则
ACM/ICPC
题目
9
开始于
2022-7-30 13:00
结束于
2022-7-30 18:00
持续时间
5 小时
主持人
参赛人数
26