题目描述
给定一个长度为 n 的序列 s 和一个正整数 k,其中 si∈{−1,1},即 si 为 −1 或 1。
我们称一个长度为 n 的序列 v 是 Yummy 的,当且仅当序列 v 中的每个元素均为不大于 k 的非负整数,且满足:
i=1∑nsi⋅vi⋅ki=0
即对于所有不大于 n 的正整数 i,si⋅vi⋅ki 之和为 0。
你需要求出不同的长度为 n 的 Yummy 的序列 v 的数量。其中,我们称两个长度均为 m 的序列 a,b 是不同的,当且仅当存在至少一个不大于 m 的正整数 i 满足 ai=bi。
由于答案可能很大,所以你只需要求出答案对 998244353 取模的结果。
输入格式
本题有多组测试数据。
输入文件的第一行输入一个正整数 T(1≤T≤104) 表示测试数据组数。
接下来,对于每一组测试数据:
第一行输入两个正整数 n(2≤n≤5×105)和 k(2≤k≤109)。
第二行输入 n 个整数 si(si∈{−1,1})。
保证对于单个测试点,所有 n 的和不超过 5×105。
输出格式
对于每组测试数据,输出一行一个正整数表示答案对 998244353 取模的结果。
2
5 2
-1 1 1 -1 -1
8 100
1 1 -1 -1 -1 -1 1 -1
5
16