#P16916. [JLCPC 2026] 计树

[JLCPC 2026] 计树

题目描述

给出一个整数 KK,以及两个长度为 2K12^K - 1 的数组 a1,a2,,a2K1a_1, a_2, \ldots, a_{2^K - 1}b1,b2,,b2K1b_1, b_2, \ldots, b_{2^K - 1}

有一个大小为 2K2^K 的无向完全图 GG,点编号范围为 02K10 \sim 2^K - 1

对于图 GG 的一棵生成树 TT,定义

$$\begin{aligned} A(T) &= \prod_{(u,v)\in T} a_{u\oplus v},\\ B(T) &= \sum_{(u,v)\in T} b_{u\oplus v},\\ C(T) &= \bigoplus_{(u,v)\in T} (u\oplus v). \end{aligned}$$

你需要对于每个 0x<2K0 \le x < 2^K,求出

$$\left(\sum_{\substack{T\\ C(T)=x}} A(T)B(T)^p\right) \bmod 998244353,$$

其中 pp 为一个给定的常数。

规定 00=10^0=1

无向完全图表示任意两个不同点之间都有一条无向边;生成树表示选出若干条边,使得所有点连通且不存在环;符号 \oplus 表示按位异或。

输入格式

第一行包含两个整数 K,pK, p1K161 \le K \le 160p50 \le p \le 5)。

接下来一行包含 2K12^K - 1 个整数,第 ii 个整数表示 aia_i0ai<9982443530 \le a_i < 998244353)。

接下来一行包含 2K12^K - 1 个整数,第 ii 个整数表示 bib_i0bi<9982443530 \le b_i < 998244353)。

输出格式

一行包含 2K2^K 个整数,第 ii 个整数表示 x=i1x = i - 1 时的答案。

2 0
1 1 1
1 2 3
4 4 4 4
2 2
2 3 5
1 4 2
5880 5416 10464 9640